CultriX commited on
Commit
20cc1bb
1 Parent(s): a4732ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -80,7 +80,7 @@ def update_selection(evt: gr.SelectData, width, height):
80
  height,
81
  )
82
 
83
- @spaces.GPU(duration=60)
84
  def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
85
  if torch.cuda.is_available():
86
  pipe.to("cuda")
@@ -196,7 +196,7 @@ def add_custom_lora(custom_lora):
196
  def remove_custom_lora():
197
  return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
198
 
199
- @spaces.GPU(duration=60)
200
  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)):
201
  if selected_index is None:
202
  raise gr.Error("You must select a LoRA before proceeding.")
 
80
  height,
81
  )
82
 
83
+ @spaces.GPU(duration=45)
84
  def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
85
  if torch.cuda.is_available():
86
  pipe.to("cuda")
 
196
  def remove_custom_lora():
197
  return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
198
 
199
+ @spaces.GPU(duration=45)
200
  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)):
201
  if selected_index is None:
202
  raise gr.Error("You must select a LoRA before proceeding.")