CultriX commited on
Commit
8dc2604
·
verified ·
1 Parent(s): 5804842

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2158,7 +2158,7 @@ def update_selection(evt: gr.SelectData, width, height):
2158
  height,
2159
  )
2160
 
2161
- @spaces.GPU(duration=60)
2162
  def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
2163
  pipe.to("cuda")
2164
  generator = torch.Generator(device="cuda").manual_seed(seed)
@@ -2195,7 +2195,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
2195
  ).images[0]
2196
  return final_image
2197
 
2198
- @spaces.GPU(duration=60)
2199
  def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
2200
  if selected_index is None:
2201
  raise gr.Error("You must select a LoRA before proceeding.🧨")
 
2158
  height,
2159
  )
2160
 
2161
+ @spaces.GPU(duration=45)
2162
  def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
2163
  pipe.to("cuda")
2164
  generator = torch.Generator(device="cuda").manual_seed(seed)
 
2195
  ).images[0]
2196
  return final_image
2197
 
2198
+ @spaces.GPU(duration=45)
2199
  def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
2200
  if selected_index is None:
2201
  raise gr.Error("You must select a LoRA before proceeding.🧨")