ysharma HF staff commited on
Commit
60bca8f
·
1 Parent(s): 0417f88
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,7 +53,7 @@ def accelerate_train_lora(steps):
53
  --lr_warmup_steps=0 \
54
  --max_train_steps={int(steps)}') #,shell=True) #30000
55
  print("*********** completing accelerate_train_lora ***********")
56
- return
57
 
58
  with gr.Blocks() as demo:
59
  with gr.Row():
@@ -67,7 +67,7 @@ with gr.Blocks() as demo:
67
  with gr.Row():
68
  in_prompt = gr.Textbox(label="Enter a prompt for fine-tuned LORA model", visible=True)
69
  out_image = gr.Image(label="Image generated by LORA model")
70
- out_file = gr.File(label="Lora trained model weights")
71
  b1.click(fn = accelerate_train_lora, inputs=in_steps, outputs=out_file)
72
  b2.click(fn = monkeypatching, inputs=[in_alpha, in_prompt], outputs=out_image)
73
 
 
53
  --lr_warmup_steps=0 \
54
  --max_train_steps={int(steps)}') #,shell=True) #30000
55
  print("*********** completing accelerate_train_lora ***********")
56
+ return "./output_example/lora_weight.pt"
57
 
58
  with gr.Blocks() as demo:
59
  with gr.Row():
 
67
  with gr.Row():
68
  in_prompt = gr.Textbox(label="Enter a prompt for fine-tuned LORA model", visible=True)
69
  out_image = gr.Image(label="Image generated by LORA model")
70
+ out_file = gr.File(label="Lora trained model weights", )
71
  b1.click(fn = accelerate_train_lora, inputs=in_steps, outputs=out_file)
72
  b2.click(fn = monkeypatching, inputs=[in_alpha, in_prompt], outputs=out_image)
73