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
|