Update app.py

app.py

@@ -14,9 +14,10 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, 2000)
     return seed
 
-device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
-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", device=device)
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=dtype).to(device)
+pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin", device=device, dtype=dtype)
 
 @spaces.GPU()
 def create_image(image_pil,