ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-29 18:51:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changed model download from HF to fp16

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-08 07:57:19 -06:00
parent ce4f9fe02a
commit cb91b0d6da
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/stable_diffusion_model.py
View File

@@ -208,7 +208,8 @@ class StableDiffusion:
device=self.device_torch, device=self.device_torch,
load_safety_checker=False, load_safety_checker=False,
requires_safety_checker=False, requires_safety_checker=False,
safety_checker=False safety_checker=False,
variant="fp16"
).to(self.device_torch) ).to(self.device_torch)
else: else:
pipe = pipln.from_single_file( pipe = pipln.from_single_file(