Update app.py

app.py

@@ -15,7 +15,7 @@ from diffusers import AutoencoderKL, EulerDiscreteScheduler
 from huggingface_hub import snapshot_download
 import spaces 
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda"
 root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 ckpt_dir = f'{root_dir}/weights/Kolors'
 
@@ -35,7 +35,7 @@ image_encoder = CLIPVisionModelWithProjection.from_pretrained(
 ).to(dtype=torch.float16, device=device)
 
 ip_img_size = 336
-clip_image_processor = CLIPImageProcessor(size=ip_img_size, crop_size=ip_img_size)
+clip_image_processor = CLIPImageProcessor(size=ip_img_size, crop_size=ip_img_size).to(device)
 
 pipe = StableDiffusionXLPipeline(
     vae=vae,
@@ -46,9 +46,9 @@ pipe = StableDiffusionXLPipeline(
     image_encoder=image_encoder,
     feature_extractor=clip_image_processor,
     force_zeros_for_empty_prompt=False
-)
+).to(device)
 
-pipe = pipe.to(device)
+#pipe = pipe.to(device)
 #pipe.enable_model_cpu_offload()
 
 if hasattr(pipe.unet, 'encoder_hid_proj'):