Spaces:
Sleeping
Sleeping
Reduce GPU duration to 30s (saves quota)
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ async def homepage():
|
|
| 63 |
return f.read()
|
| 64 |
|
| 65 |
|
| 66 |
-
@spaces.GPU
|
| 67 |
def run_inference(prompt, width, height, seed):
|
| 68 |
"""Run the diffusion pipeline on GPU."""
|
| 69 |
generator = torch.Generator("cuda").manual_seed(seed)
|
|
|
|
| 63 |
return f.read()
|
| 64 |
|
| 65 |
|
| 66 |
+
@spaces.GPU(duration=30)
|
| 67 |
def run_inference(prompt, width, height, seed):
|
| 68 |
"""Run the diffusion pipeline on GPU."""
|
| 69 |
generator = torch.Generator("cuda").manual_seed(seed)
|