1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-24 20:59:27 +03:00
Files
Ricardo-M-L e1db6d1c7d Fix AttributeError in onnxruntime train_unconditional (args.report_to → args.logger) (#13524)
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`.
2026-06-01 09:23:34 -07:00
..
2026-01-10 11:11:15 -08:00
2023-12-11 10:55:28 -08:00

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.