add lora_rank for lora stable diffusion

2026-07-13 13:59:40 +02:00 · 2023-07-24 19:32:04 +08:00
parent eb24e23d19
commit bc93e2dc96
1 changed files with 1 additions and 2 deletions
--- a/examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py
+++ b/examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py
@@ -22,8 +22,7 @@ class StableDiffusionLoraArguments(TrainingArgs):
    lora_rank: int = field(
        default=4, metadata={
            'help': 'The rank size of lora intermediate linear.',
-        }
-    )
+        })


 training_args = StableDiffusionLoraArguments(