soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages 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
Show Stats 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')

21
modules_forge/initialization.py
View File

@ -1,3 +1,4 @@
import os
import sys
@ -49,4 +50,24 @@ def initialize_forge():
import modules_forge.patch_basic
modules_forge.patch_basic.patch_all_basics()
from modules_forge.shared import diffusers_dir
if 'TRANSFORMERS_CACHE' not in os.environ:
os.environ['TRANSFORMERS_CACHE'] = diffusers_dir
if 'HF_HOME' not in os.environ:
os.environ['HF_HOME'] = diffusers_dir
if 'HF_DATASETS_CACHE' not in os.environ:
os.environ['HF_DATASETS_CACHE'] = diffusers_dir
if 'HUGGINGFACE_HUB_CACHE' not in os.environ:
os.environ['HUGGINGFACE_HUB_CACHE'] = diffusers_dir
if 'HUGGINGFACE_ASSETS_CACHE' not in os.environ:
os.environ['HUGGINGFACE_ASSETS_CACHE'] = diffusers_dir
if 'HF_HUB_CACHE' not in os.environ:
os.environ['HF_HUB_CACHE'] = diffusers_dir
return

3
modules_forge/shared.py
View File

@ -10,6 +10,9 @@ os.makedirs(controlnet_dir, exist_ok=True)
preprocessor_dir = os.path.join(models_path, 'ControlNetPreprocessor')
os.makedirs(preprocessor_dir, exist_ok=True)
diffusers_dir = os.path.join(models_path, 'diffusers')
os.makedirs(diffusers_dir, exist_ok=True)
supported_preprocessors = {}
supported_control_models = []