diffusers
326de419 - Trivial fix for undefined symbol in train_dreambooth.py (#1598)

Commit
3 years ago
Trivial fix for undefined symbol in train_dreambooth.py (#1598) easy fix for undefined name in train_dreambooth.py import_model_class_from_model_name_or_path loads a pretrained model and refers to args.revision in a context where args is undefined. I modified the function to take revision as an argument and modified the invocation of the function to pass in the revision from args. Seems like this was caused by a cut and paste.
Author
Parents
Loading