diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

Fix AttributeError in train_lcm_distill_lora_sdxl_wds.py

Open jainalphin opened this issue 1 year ago • 1 comments

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

jainalphin avatar May 12 '24 09:05 jainalphin

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.