Changed model download from HF to fp16

This commit is contained in:
Jaret Burkett committed 2023-09-08 07:57:19 -06:00
1 parent ce4f9fe02a
commit cb91b0d6da
1 file changed
+2 -1
+2 -1
View File
@@ -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(