Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -654,7 +654,7 @@ def create_demo():
|
|
| 654 |
)
|
| 655 |
|
| 656 |
# Set up event handlers
|
| 657 |
-
@spaces.GPU(duration=
|
| 658 |
def generate(prompt, guidance_scale, num_steps):
|
| 659 |
try:
|
| 660 |
# Ensure PyTorch3D works with CUDA
|
|
|
|
| 654 |
)
|
| 655 |
|
| 656 |
# Set up event handlers
|
| 657 |
+
@spaces.GPU(duration=90) # Reduced duration to 20 seconds
|
| 658 |
def generate(prompt, guidance_scale, num_steps):
|
| 659 |
try:
|
| 660 |
# Ensure PyTorch3D works with CUDA
|