Changed model download from HF to fp16

2023-09-08 07:57:19 -06:00 · 2023-09-08 07:57:19 -06:00 · cb91b0d6da
parent ce4f9fe02a
commit cb91b0d6da
1 changed files with 2 additions and 1 deletions
--- 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(