Spaces:
Configuration error
Configuration error
update example lora weights
Browse files
app.py
CHANGED
@@ -58,7 +58,8 @@ def accelerate_train_lora(steps):
|
|
58 |
--lr_warmup_steps=0 \
|
59 |
--max_train_steps={int(steps)}') #,shell=True) #30000
|
60 |
print("*********** completing accelerate_train_lora ***********")
|
61 |
-
|
|
|
62 |
|
63 |
with gr.Blocks() as demo:
|
64 |
gr.Markdown("""<h1><center>LORA - Low-rank Adaptation for Fast Text-to-Image Diffusion Fine-tuning</center></h1>
|
|
|
58 |
--lr_warmup_steps=0 \
|
59 |
--max_train_steps={int(steps)}') #,shell=True) #30000
|
60 |
print("*********** completing accelerate_train_lora ***********")
|
61 |
+
lora_trained_weights = "./output_example/lora_weight.pt"
|
62 |
+
return lora_trained_weights #"./output_example/lora_weight.pt"
|
63 |
|
64 |
with gr.Blocks() as demo:
|
65 |
gr.Markdown("""<h1><center>LORA - Low-rank Adaptation for Fast Text-to-Image Diffusion Fine-tuning</center></h1>
|