generate_image_lowmem GPU management Update
Browse files
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=
|
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("
|
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(
|