YiftachEde commited on
Commit
a0a4b6f
·
verified ·
1 Parent(s): 9dbb75f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -539,7 +539,7 @@ def create_demo():
539
  )
540
 
541
  # Set up event handlers
542
- @spaces.GPU(duration=80) # 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
 
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