Spaces:
Running
on
Zero
Running
on
Zero
chore: set duration
Browse files- diffusion.py +2 -2
diffusion.py
CHANGED
@@ -94,10 +94,10 @@ def image_generation_config_ui():
|
|
94 |
class ImageGenerator:
|
95 |
# pipe: StableDiffusionXLPipeline
|
96 |
|
97 |
-
def __init__(self
|
98 |
pass
|
99 |
|
100 |
-
@spaces.GPU()
|
101 |
def generate(
|
102 |
self,
|
103 |
prompt: str,
|
|
|
94 |
class ImageGenerator:
|
95 |
# pipe: StableDiffusionXLPipeline
|
96 |
|
97 |
+
def __init__(self):
|
98 |
pass
|
99 |
|
100 |
+
@spaces.GPU(duration=30)
|
101 |
def generate(
|
102 |
self,
|
103 |
prompt: str,
|