aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-20 10:14:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix another pydantic 2 migration error

Browse Source
This commit is contained in:
lucidrains
2023-10-18 21:07:47 -07:00
parent dab2f74650
commit b6fecae91a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_decoder.py
View File

@@ -636,7 +636,7 @@ def initialize_training(config: TrainDecoderConfig, config_path):
inference_device=accelerator.device, inference_device=accelerator.device,
evaluate_config=config.evaluate, evaluate_config=config.evaluate,
condition_on_text_encodings=conditioning_on_text, condition_on_text_encodings=conditioning_on_text,
**config.train.dict(), **config.train.model_dump(),
) )
# Create a simple click command line interface to load the config and start the training # Create a simple click command line interface to load the config and start the training