Update app.py

app.py

@@ -28,8 +28,13 @@ pipe.load_lora_weights(f'{pipeline_path}/pytorch_lora_weights.safetensors')
 pipe.fuse_lora()
 pipe.unload_lora_weights()
 pipe.force_zeros_for_empty_prompt = False
+
+pipe.to("cuda")
+
 pipe = EllaXLPipeline(pipe,f'{pipeline_path}/pytorch_model.bin')
 
+
+
 def tocuda():
     pipe.pipe.vae.to('cuda')
     pipe.t5_encoder.to('cuda')
@@ -61,8 +66,8 @@ def tocuda():
 @spaces.GPU(enable_queue=True)
 def infer(prompt,negative_prompt,seed,resolution, steps):
     
-    if 'cuda' not in pipe.pipe.device.type:
-        tocuda()
+    # if 'cuda' not in pipe.pipe.device.type:
+    #     tocuda()
         
     print(f"""
     —/n