This repository was archived by the owner on Feb 7, 2025. It is now read-only.
This repository was archived by the owner on Feb 7, 2025. It is now read-only.
Issue when sampling the VQVAE+Transformer from 2d_vqvae_transformer_tutorial #320
Description
Metadata
Metadata
Assignees
Labels
No labels
In 2d_vqvae_transformer_tutorial.ipy, I fixed the num_tokens in the training in #303 . But I missed the sampling code. In
GenerativeModels/tutorials/generative/2d_vqvae_transformer/2d_vqvae_transformer_tutorial.py
Line 367 in 6a9cc93
it should be
probs[:, vqvae_model.num_embeddings] = 0