Surn commited on
Commit
48ec05f
·
1 Parent(s): 71bc30e

generate_image_lowmem GPU management Update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -305,7 +305,7 @@ class Condition(object):
305
  # def generate_image(pipe, generate_params, progress=gr.Progress(track_tqdm=True)):
306
  # return pipe(**generate_params)
307
 
308
- @spaces.GPU(duration=140, progress=gr.Progress(track_tqdm=True))
309
  def generate_image_lowmem(
310
  text,
311
  neg_prompt=None,
@@ -1021,7 +1021,7 @@ with gr.Blocks(css_paths="style_20250128.css", title=title, theme='Surn/beeuty',
1021
  )
1022
 
1023
  with gr.Row():
1024
- with gr.Accordion("Generative AI", open = False):
1025
  with gr.Row():
1026
  with gr.Column():
1027
  model_options = gr.Dropdown(
 
305
  # def generate_image(pipe, generate_params, progress=gr.Progress(track_tqdm=True)):
306
  # return pipe(**generate_params)
307
 
308
+ @spaces.GPU(duration=200, progress=gr.Progress(track_tqdm=True))
309
  def generate_image_lowmem(
310
  text,
311
  neg_prompt=None,
 
1021
  )
1022
 
1023
  with gr.Row():
1024
+ with gr.Accordion("Generate AI Image (click here)", open = False):
1025
  with gr.Row():
1026
  with gr.Column():
1027
  model_options = gr.Dropdown(