add lora_rank for lora stable diffusion

2026-02-24 12:10:09 +01:00 · 2023-07-24 19:43:20 +08:00
parent bc93e2dc96
commit 426f55d57b
1 changed files with 1 additions and 1 deletions
--- a/examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py
+++ b/examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py
@@ -18,7 +18,7 @@ class StableDiffusionLoraArguments(TrainingArgs):
        default='dog', metadata={
            'help': 'The pipeline prompt.',
        })
-    
+
    lora_rank: int = field(
        default=4, metadata={
            'help': 'The rank size of lora intermediate linear.',