Fix typo in train_diffusion_orpo_sdxl_lora_wds.py (#11541)

ddd0cfb4 · Abdellah Oumida · GitHub · 4f438de3 · ddd0cfb4
Unverified Commit ddd0cfb4 authored May 13, 2025 by Abdellah Oumida Committed by GitHub May 12, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/research_projects/diffusion_orpo/train_diffusion_orpo_sdxl_lora_wds.py ...ects/diffusion_orpo/train_diffusion_orpo_sdxl_lora_wds.py +1 -1

No files found.
--- a/examples/research_projects/diffusion_orpo/train_diffusion_orpo_sdxl_lora_wds.py
+++ b/examples/research_projects/diffusion_orpo/train_diffusion_orpo_sdxl_lora_wds.py
@@ -812,7 +812,7 @@ def main(args):

    if args.scale_lr:
        args.learning_rate = (
-            args.learning_rat * args.gradient_accumulation_steps * args.per_gpu_batch_size * accelerator.num_processes
+            args.learning_rate * args.gradient_accumulation_steps * args.per_gpu_batch_size * accelerator.num_processes
        )

    # Use 8-bit Adam for lower memory usage or to fine-tune the model in 16GB GPUs