sd: load single file model using local config

This commit is contained in:
Qing
2024-01-30 13:19:13 +08:00
parent 934b0fc455
commit e5f71ae535
12 changed files with 352 additions and 2 deletions

View File

@@ -5,6 +5,7 @@ from loguru import logger
from .base import DiffusionInpaintModel
from .helper.cpu_text_encoder import CPUTextEncoderWrapper
from .original_sd_configs import get_config_files
from .utils import (
handle_from_pretrained_exceptions,
get_torch_dtype,
@@ -51,6 +52,7 @@ class SD(DiffusionInpaintModel):
self.model_id_or_path,
dtype=torch_dtype,
load_safety_checker=not disable_nsfw_checker,
config_files=get_config_files(),
**model_kwargs,
)
else: