Spaces:

ysharma
/

Low-rank-Adaptation

Configuration error

ysharma HF Staff commited on Dec 17, 2022

Commit

b63efd7

1 Parent(s): 0d33cea

update lr

Files changed (1) hide show

app.py CHANGED Viewed

@@ -70,8 +70,8 @@ def accelerate_train_lora(steps, images, in_prompt):
       --resolution=512 \
       --train_batch_size=1 \
       --gradient_accumulation_steps=1 \
-      --learning_rate=1e-4 \
-      --learning_rate_text=5e-5 \
       --color_jitter \
       --lr_scheduler='constant' \
       --lr_warmup_steps=0 \

       --resolution=512 \
       --train_batch_size=1 \
       --gradient_accumulation_steps=1 \
+      --learning_rate='1e-4' \
+      --learning_rate_text='5e-5' \
       --color_jitter \
       --lr_scheduler='constant' \
       --lr_warmup_steps=0 \