jiuface commited on
Commit
5f10541
·
1 Parent(s): 7e0a1d6

support multi lora

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -85,7 +85,7 @@ def upload_image_to_r2(image, account_id, access_key, secret_key, bucket_name):
85
  return image_file
86
 
87
 
88
- @spaces.GPU
89
  def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
90
  pipe.to("cuda")
91
  generator = torch.Generator(device="cuda").manual_seed(seed)
@@ -104,7 +104,7 @@ def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
104
  progress(99, "Generate success!")
105
  return generate_image
106
 
107
-
108
  def run_lora(prompt, cfg_scale, steps, lora_strings, randomize_seed, seed, width, height, lora_scale, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
109
 
110
 
@@ -167,7 +167,7 @@ with gr.Blocks(css=css) as demo:
167
  with gr.Row():
168
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
169
  randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
170
- lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=0.95)
171
 
172
  with gr.Row():
173
  width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
 
85
  return image_file
86
 
87
 
88
+
89
  def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
90
  pipe.to("cuda")
91
  generator = torch.Generator(device="cuda").manual_seed(seed)
 
104
  progress(99, "Generate success!")
105
  return generate_image
106
 
107
+ @spaces.GPU
108
  def run_lora(prompt, cfg_scale, steps, lora_strings, randomize_seed, seed, width, height, lora_scale, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
109
 
110
 
 
167
  with gr.Row():
168
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
169
  randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
170
+ lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=1, step=0.01, value=0.5)
171
 
172
  with gr.Row():
173
  width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)