Update app.py

app.py

@@ -27,7 +27,7 @@ pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3", torch_dtype=torch.fl
 pipe.load_lora_weights(f'{pipeline_path}/pytorch_lora_weights.safetensors')
 pipe.fuse_lora()
 pipe.unload_lora_weights()
-pipe.to("cuda")
+# pipe.to("cuda")
 pipe.force_zeros_for_empty_prompt = False
 pipe = EllaXLPipeline(pipe,f'{pipeline_path}/pytorch_model.bin')
 
@@ -61,7 +61,7 @@ def infer(prompt,negative_prompt,seed,resolution, steps):
     
     # generator = torch.Generator("cuda").manual_seed(555)
     t=time.time()
-
+    pipe.to("cuda")
     if seed=="-1":
         generator=None
     else: