From 426f55d57b8932aeb41aa1941f2172475542e8ca Mon Sep 17 00:00:00 2001 From: XDUWQ <1300964705@qq.com> Date: Mon, 24 Jul 2023 19:43:20 +0800 Subject: [PATCH] add lora_rank for lora stable diffusion --- .../stable_diffusion/lora/finetune_stable_diffusion_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 e15aa444..325cc7ae 100644 --- 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.',