diffusers
diffusers copied to clipboard
Fix AttributeError in train_lcm_distill_lora_sdxl_wds.py
What does this PR do?
This pull request fixes an AttributeError in train_lcm_distill_lora_sdxl_wds.py by updating the code to use args.pretrained_teacher_model instead of args.pretrained_model_name_or_path.
This fix resolves the error:
File "/diffusers/examples/consistency_distillation/train_lcm_distill_lora_sdxl_wds.py", line 1530, in <module>
main(args)
File "/diffusers/examples/consistency_distillation/train_lcm_distill_lora_sdxl_wds.py", line 1361, in main
if torch.backends.mps.is_available() or "playground" in args.pretrained_model_name_or_path:
AttributeError: 'Namespace' object has no attribute 'pretrained_model_name_or_path'. Did you mean: 'pretrained_vae_model_name_or_path'?
Before submitting
- [x] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
- [x] Did you read the contributor guideline?
Who can review?
- Training examples: @sayakpaul
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.