fix a bug of name error (#179)

This commit is contained in:
yytdfc
2022-06-29 22:16:44 +08:00
committed by GitHub
parent 6a11b9678b
commit 86109646e3

View File

@@ -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