search

lucidrains / imagen-pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
MIT License
8.11k stars 768 forks source link

Fixes #274

Closed TheFusion21 closed 2 years ago

TheFusion21 commented 2 years ago

Moved dataloader out of init again to reduce args Moved validate_and_set_unet_being_trained into own method as prepare function This maked add_train_dataloader, add_train_dataset etc usable again and still works with deepspeed

gowdygamble commented 2 years ago

cool, better fix than mine! Thanks