Update app.py

app.py

@@ -15,7 +15,6 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
-torch.cuda.set_device(device)
 pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16).to(device)
 pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
 
@@ -31,7 +30,7 @@ def create_image(image_pil,
                  target="Load only style blocks",
                  ):
 
-    
+    torch.cuda.set_device(device)
     if target !="Load original IP-Adapter":
         if target=="Load only style blocks":
             scale = {