Update app.py

app.py

@@ -35,11 +35,11 @@ pipe = EllaXLPipeline(pipe,f'{pipeline_path}/pytorch_model.bin')
 
 
 
-def tocuda():
-    pipe.pipe.vae.to('cuda')
-    pipe.t5_encoder.to('cuda')
-    pipe.pipe.unet.unet.to('cuda')
-    pipe.pipe.unet.ella.to('cuda')
+# def tocuda():
+#     pipe.pipe.vae.to('cuda')
+#     pipe.t5_encoder.to('cuda')
+#     pipe.pipe.unet.unet.to('cuda')
+#     pipe.pipe.unet.ella.to('cuda')
 
 
 # print("Optimizing BRIA-2.3-T5 - this could take a while")
@@ -63,7 +63,7 @@ def tocuda():
 
 # print(f"Optimizing finished successfully after {time.time()-t} secs")
 
-@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True, duration = 120)
 def infer(prompt,negative_prompt,seed,resolution, steps):
     
     # if 'cuda' not in pipe.pipe.device.type: