mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-23 09:09:01 +03:00
Fix AttributeError in onnxruntime train_unconditional The ORT example `examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py` defines `--logger` (argparse line 196) but references `args.report_to` in two places: * Line 280: `if args.report_to == "wandb" and args.hub_token is not None:` * Line 294: `log_with=args.report_to,` Since there is no `--report_to` flag on the parser, both references crash with `AttributeError: 'Namespace' object has no attribute 'report_to'` as soon as `main()` starts — the script cannot run at all. The sibling non-ORT script `examples/unconditional_image_generation/train_unconditional.py` uses `args.logger` consistently; this PR aligns the ORT variant with that behavior, and updates the error message to refer to `--logger=wandb` instead of the non-existent `--report_to=wandb`.
Diffusers examples with ONNXRuntime optimizations
This research project is not actively maintained by the diffusers team. For any questions or comments, please contact Prathik Rao (prathikr), Sunghoon Choi (hanbitmyths), Ashwini Khade (askhade), or Peng Wang (pengwa) on github with any questions.
This aims to provide diffusers examples with ONNXRuntime optimizations for training/fine-tuning unconditional image generation, text to image, and textual inversion. Please see individual directories for more details on how to run each task using ONNXRuntime.