Issue when sampling the VQVAE+Transformer from 2d_vqvae_transformer_tutorial

[Warvito]

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

probs[:, vqvae_model.num_embeddings - 1] = 0

it should be
probs[:, vqvae_model.num_embeddings] = 0