Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -539,7 +539,7 @@ def create_demo():
|
|
| 539 |
)
|
| 540 |
|
| 541 |
# Set up event handlers
|
| 542 |
-
@spaces.GPU(duration=
|
| 543 |
def generate(prompt, guidance_scale, num_steps):
|
| 544 |
try:
|
| 545 |
torch.cuda.empty_cache() # Clear GPU memory before starting
|
|
|
|
| 539 |
)
|
| 540 |
|
| 541 |
# Set up event handlers
|
| 542 |
+
@spaces.GPU(duration=100) # Add GPU decorator to the generate function
|
| 543 |
def generate(prompt, guidance_scale, num_steps):
|
| 544 |
try:
|
| 545 |
torch.cuda.empty_cache() # Clear GPU memory before starting
|