Spaces:
Running
on
Zero
Running
on
Zero
Julian Bilcke
commited on
Commit
·
cda2b7d
1
Parent(s):
4f5fb48
update timeout
Browse files- app_gradio.py +3 -2
app_gradio.py
CHANGED
|
@@ -319,8 +319,9 @@ generation_system = GenerationSystem(ckpt_path=ckpt_path, device=device, offload
|
|
| 319 |
print("Model loaded successfully!")
|
| 320 |
|
| 321 |
|
| 322 |
-
# GPU-decorated generation function with
|
| 323 |
-
|
|
|
|
| 324 |
def generate_scene(
|
| 325 |
image_prompt,
|
| 326 |
text_prompt,
|
|
|
|
| 319 |
print("Model loaded successfully!")
|
| 320 |
|
| 321 |
|
| 322 |
+
# GPU-decorated generation function with 60-second budget
|
| 323 |
+
# (necessary fro the initialization and loading of everything)
|
| 324 |
+
@GPU(duration=60)
|
| 325 |
def generate_scene(
|
| 326 |
image_prompt,
|
| 327 |
text_prompt,
|