Update app.py

app.py

@@ -16,8 +16,6 @@ dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32).to(device)
 pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
 
-print(device)
-
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, 2000)
@@ -34,7 +32,7 @@ def create_image(image_pil,
                  seed,
                  target="Load only style blocks",
                  ):
-    print(device)
+    
     if target !="Load original IP-Adapter":
         if target=="Load only style blocks":
             scale = {
@@ -53,8 +51,8 @@ def create_image(image_pil,
 
     style_image = load_image(image_pil)
     generator = torch.Generator(device=device).manual_seed(randomize_seed_fn(seed, True))
-    torch.cuda.set_device(device)
-    print(device)
+    
+    
     image = pipeline(
         prompt=prompt,
         ip_adapter_image=style_image,