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.',