Changed model download from HF to fp16
This commit is contained in:
1 parent
ce4f9fe02a
commit
cb91b0d6da
1 file changed
+2
-1
@@ -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(
|
||||
|
||||
Reference in new issue
Block a user