github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #12 from bendeguzvaradi/main

Browse Source 
Bug/Safety checker to None
This commit is contained in:
Jaret Burkett
2023-09-14 15:31:09 -06:00
committed by GitHub
parent bee0b6a235 3d387103cd
commit 3eb3535683
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/stable_diffusion_model.py
View File

@@ -212,7 +212,7 @@ 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=None
).to(self.device_torch) ).to(self.device_torch)
else: else:
pipe = pipln.from_single_file( pipe = pipln.from_single_file(