From 86109646e3a1fa5cf3ea10d939fe1441302845a9 Mon Sep 17 00:00:00 2001 From: yytdfc Date: Wed, 29 Jun 2022 22:16:44 +0800 Subject: [PATCH] fix a bug of name error (#179) --- dalle2_pytorch/train_configs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dalle2_pytorch/train_configs.py b/dalle2_pytorch/train_configs.py index 1f9ccec..a333637 100644 --- a/dalle2_pytorch/train_configs.py +++ b/dalle2_pytorch/train_configs.py @@ -289,7 +289,7 @@ class TrainDecoderConfig(BaseModel): # Then something else errored and we should just pass through return values - using_text_encodings = any([unet.cond_on_text_encodings for unet in decoder_config.unets]) + using_text_embeddings = any([unet.cond_on_text_encodings for unet in decoder_config.unets]) using_clip = exists(decoder_config.clip) img_emb_url = data_config.img_embeddings_url text_emb_url = data_config.text_embeddings_url