PixelCNNSampler sample function fix (#148)

* fix pixelcnnsampler sample func

* ignore tests/dummy_output_dir

* Update .gitignore

.gitignore

@@ -23,6 +23,7 @@ examples/notebooks/models_training/my_model/
 examples/showcases copy/*
 examples/notebooks/models_training/generate.py
 dummy_output_dir/*
+tests/dummy_output_dir/
 examples/notebooks/models_training/my_model/*
 examples/notebooks/test.ipynb
 examples/notebooks/dummy_output_dir/*

src/pythae/samplers/pixelcnn_sampler/pixelcnn_sampler.py

@@ -203,7 +203,10 @@ def sample(
                     probs = F.softmax(out[:, :, k, l], dim=-1)
                     z[:, :, k, l] = torch.multinomial(probs, 1).float()
 
-            z_quant = self.model.quantizer.embeddings(z.reshape(z.shape[0], -1).long())
+            if isinstance(self.model.quantizer.embeddings, torch.Tensor):
+                z_quant = self.model.quantizer.embeddings[z.reshape(z.shape[0], -1).long()]
+            else:
+                z_quant = self.model.quantizer.embeddings(z.reshape(z.shape[0], -1).long())
 
             if self.needs_reshape:
                 z_quant = z_quant.reshape(z.shape[0], -1)