duration

app.py

@@ -455,7 +455,7 @@ def create_demo(model):
     return demo
 
 
-@spaces.GPU
+@spaces.GPU(duration=180)
 def generate(
     images, 
     prompts, 

visualcloze.py

@@ -144,6 +144,7 @@ class VisualClozeModel:
         self.grid_h = h
         self.grid_w = w
     
+    @torch.no_grad
     def upsampling(self, image, target_size, cfg, upsampling_steps, upsampling_noise, generator, content_prompt):
         content_instruction = [
             "The content of the last image in the final row is: ",