cleanup

2025-12-20 18:24:19 +01:00 · 2022-04-13 18:23:24 -07:00
parent 0692f1699f
commit 14ddbc159c
3 changed files with 2 additions and 3 deletions
--- a/dalle2_pytorch/dalle2_pytorch.py
+++ b/dalle2_pytorch/dalle2_pytorch.py
@@ -877,7 +877,6 @@ class DALLE2(nn.Module):
            text = [text] if not isinstance(text, (list, tuple)) else text
            text = tokenizer.tokenize(text).to(device)

-        print(text.shape, type(text))
        image_embed = self.prior.sample(text, num_samples_per_batch = self.prior_num_samples)
        images = self.decoder.sample(image_embed)
        return images