From fac63c61bce47610469812b6591a876c41e430f4 Mon Sep 17 00:00:00 2001 From: Aidan Dempster Date: Sat, 2 Jul 2022 11:56:03 -0400 Subject: [PATCH] Fixed variable naming issue (#183) --- dalle2_pytorch/train_configs.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dalle2_pytorch/train_configs.py b/dalle2_pytorch/train_configs.py index c1902b6..8e7560f 100644 --- a/dalle2_pytorch/train_configs.py +++ b/dalle2_pytorch/train_configs.py @@ -346,17 +346,17 @@ class TrainDecoderConfig(BaseModel): img_emb_url = data_config.img_embeddings_url text_emb_url = data_config.text_embeddings_url - if using_text_encodings: + if using_text_embeddings: # Then we need some way to get the embeddings assert using_clip or exists(text_emb_url), 'If text conditioning, either clip or text_embeddings_url must be provided' if using_clip: - if using_text_encodings: + if using_text_embeddings: assert not exists(text_emb_url) or not exists(img_emb_url), 'Loaded clip, but also provided text_embeddings_url and img_embeddings_url. This is redundant. Remove the clip model or the text embeddings' else: assert not exists(img_emb_url), 'Loaded clip, but also provided img_embeddings_url. This is redundant. Remove the clip model or the embeddings' if text_emb_url: - assert using_text_encodings, "Text embeddings are being loaded, but text embeddings are not being conditioned on. This will slow down the dataloader for no reason." + assert using_text_embeddings, "Text embeddings are being loaded, but text embeddings are not being conditioned on. This will slow down the dataloader for no reason." return values