Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: move censor to extras, optimize safety checker file handling

Browse Source
This commit is contained in:
Manuel Schmid
2024-05-18 01:59:15 +02:00
parent 0f78f8d8cc
commit 7568b72d9b
7 changed files with 211 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/config.py
+8
View File
@@ -685,5 +685,13 @@ def downloading_upscale_model():
)
return os.path.join(path_upscale_models, 'fooocus_upscaler_s409985e5.bin')
def downloading_safety_checker_model():
load_file_from_url(
url='https://huggingface.co/mashb1t/misc/resolve/main/stable-diffusion-safety-checker.bin',
model_dir=path_safety_checker_models,
file_name='stable-diffusion-safety-checker.bin'
)
return os.path.join(path_safety_checker_models, 'stable-diffusion-safety-checker.bin')
update_files()