dg845
fc61b531fb
* initial commit for unconditional/class-conditional consistency training script * make style * Add entry for consistency training script in community README. * Move consistency training script from community to research_projects/consistency_training * Add requirements.txt and README to research_projects/consistency_training directory. * Manually revert community README changes for consistency training. * Fix path to script after moving script to research projects. * Add option to load U-Net weights from pretrained model. --------- Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
6 lines
75 B
Plaintext
6 lines
75 B
Plaintext
accelerate>=0.16.0
|
|
torchvision
|
|
transformers>=4.25.1
|
|
ftfy
|
|
tensorboard
|
|
Jinja2 |