fix: dtype for cpu

model.py

@@ -6,6 +6,7 @@ def get_pipeline():
     from diffusers import AutoencoderTiny, AutoPipelineForImage2Image
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
     pipe = AutoPipelineForImage2Image.from_pretrained(
         "SimianLuo/LCM_Dreamshaper_v7",
@@ -13,10 +14,10 @@ def get_pipeline():
     )
     pipe.vae = AutoencoderTiny.from_pretrained(
         "madebyollin/taesd",
-        torch_dtype=torch.float16,
+        torch_dtype=torch_dtype,
         use_safetensors=True,
     )
-    pipe = pipe.to(device, dtype=torch.float16)
+    pipe = pipe.to(device, dtype=torch_dtype)
     pipe.unet.to(memory_format=torch.channels_last)
     return pipe