Update app.py

app.py

@@ -8,8 +8,8 @@ from PIL import Image
 import spaces
 
 # 🌟 Auto-detect device (CPU/GPU)
-device = "cuda" if torch.cuda.is_available() else "cpu"
-precision = torch.float16 if device == "cuda" else torch.float32
+device = "cuda"
+precision = torch.float16
 
 
 eulera_scheduler = EulerAncestralDiscreteScheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler")
@@ -21,13 +21,13 @@ controlnet = ControlNetModel.from_pretrained(
 )
 
 # when test with other base model, you need to change the vae also.
-vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=precision)
 
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     controlnet=controlnet,
     vae=vae,
-    torch_dtype=torch.float16,
+    torch_dtype=precision,
     scheduler=eulera_scheduler,
 )