From bc93e2dc96febadc6f285af1c5642c19871e1cea Mon Sep 17 00:00:00 2001
From: XDUWQ <1300964705@qq.com>
Date: Mon, 24 Jul 2023 19:32:04 +0800
Subject: [PATCH] add lora_rank for lora stable diffusion

---
 .../stable_diffusion/lora/finetune_stable_diffusion_lora.py    | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py b/examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py
index b878b5f8..e15aa444 100644
--- 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(