mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2026-02-14 17:14:38 +01:00
Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
1212f7058d |
@@ -1980,7 +1980,7 @@ class Decoder(BaseGaussianDiffusion):
|
|||||||
batch_size = image_embed.shape[0]
|
batch_size = image_embed.shape[0]
|
||||||
|
|
||||||
if exists(text) and not exists(text_encodings) and not self.unconditional:
|
if exists(text) and not exists(text_encodings) and not self.unconditional:
|
||||||
assert exist(self.clip)
|
assert exists(self.clip)
|
||||||
_, text_encodings, text_mask = self.clip.embed_text(text)
|
_, text_encodings, text_mask = self.clip.embed_text(text)
|
||||||
|
|
||||||
assert not (self.condition_on_text_encodings and not exists(text_encodings)), 'text or text encodings must be passed into decoder if specified'
|
assert not (self.condition_on_text_encodings and not exists(text_encodings)), 'text or text encodings must be passed into decoder if specified'
|
||||||
|
|||||||
Reference in New Issue
Block a user