lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-28 02:33:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

better diffusers handling

Browse Source
This commit is contained in:
lllyasviel
2024-02-03 17:16:21 -08:00
parent e3fca95012
commit a544ececfe
3 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
extensions-builtin/forge_preprocessor_marigold/scripts/preprocessor_marigold.py
View File

@@ -33,17 +33,9 @@ class PreprocessorMarigold(Preprocessor):
if self.model_patcher is not None:
return
checkpoint_path = os.path.join(preprocessor_dir, 'marigold')
if not os.path.exists(checkpoint_path):
snapshot_download(repo_id="Bingxin/Marigold",
ignore_patterns=["*.bin"],
local_dir=checkpoint_path,
local_dir_use_symlinks=False)
self.diffusers_patcher = DiffusersModelPatcher(
pipeline_class=MarigoldPipeline,
pretrained_path=checkpoint_path,
pretrained_path="Bingxin/Marigold",
enable_xformers=False,
noise_scheduler_type='DDIMScheduler')