ysharma HF Staff commited on
Commit
5345d10
·
1 Parent(s): 797e0cc

update example lora weights

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- return "./output_example/lora_weight.pt"
 
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>