diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py
index 176b77a..6f87e15 100644
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -208,7 +208,8 @@ class StableDiffusion:
                     device=self.device_torch,
                     load_safety_checker=False,
                     requires_safety_checker=False,
-                    safety_checker=False
+                    safety_checker=False,
+                    variant="fp16"
                 ).to(self.device_torch)
             else:
                 pipe = pipln.from_single_file(