hmm

app_gradio.py

@@ -319,9 +319,9 @@ generation_system = GenerationSystem(ckpt_path=ckpt_path, device=device, offload
 print("Model loaded successfully!")
 
 
-# GPU-decorated generation function with 60-second budget
-# (necessary fro the initialization and loading of everything)
-@GPU(duration=60)
+# GPU-decorated generation function with 120-second budget
+# (yes, it should be shorted.. idk why it is slow, maybe we are missing some optimizations in the dependencies, or the attention mechanism)
+@GPU(duration=120)
 def generate_scene(
     image_prompt,
     text_prompt,