From cb91b0d6da30512f4dbc1f49bd8ebd09b1eddfea Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Fri, 8 Sep 2023 07:57:19 -0600 Subject: [PATCH] Changed model download from HF to fp16 --- toolkit/stable_diffusion_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py index 176b77a3..6f87e15a 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(