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(