Update app.py

app.py

@@ -13,7 +13,7 @@ from torchvision import transforms
 
 device = torch.device('cuda:0' 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, device_map='balanced')
+pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32).to('cpu')
 pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -49,7 +49,7 @@ def create_image(image_pil,
         pipeline.set_ip_adapter_scale(scale)
 
     style_image = load_image(image_pil)
-    generator = torch.Generator(device=device).manual_seed(randomize_seed_fn(seed, True))
+    generator = torch.Generator(device='cpu').manual_seed(randomize_seed_fn(seed, True))
     torch.cuda.set_device(device)
     image = pipeline(
         prompt=prompt,
@@ -58,7 +58,7 @@ def create_image(image_pil,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
         generator=generator,
-        device=device
+        device='cuda:0'
     )
     return image