Julian Bilcke commited on
Commit
cda2b7d
·
1 Parent(s): 4f5fb48

update timeout

Browse files
Files changed (1) hide show
  1. app_gradio.py +3 -2
app_gradio.py CHANGED
@@ -319,8 +319,9 @@ generation_system = GenerationSystem(ckpt_path=ckpt_path, device=device, offload
319
  print("Model loaded successfully!")
320
 
321
 
322
- # GPU-decorated generation function with 15-second budget
323
- @GPU(duration=15)
 
324
  def generate_scene(
325
  image_prompt,
326
  text_prompt,
 
319
  print("Model loaded successfully!")
320
 
321
 
322
+ # GPU-decorated generation function with 60-second budget
323
+ # (necessary fro the initialization and loading of everything)
324
+ @GPU(duration=60)
325
  def generate_scene(
326
  image_prompt,
327
  text_prompt,