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

@@ -10,6 +10,7 @@ from iopaint.schema import InpaintRequest, ModelType
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,
@@ -41,6 +42,7 @@ class SDXL(DiffusionInpaintModel):
dtype=torch_dtype,
num_in_channels=num_in_channels,
load_safety_checker=False,
config_files=get_config_files()
)
else:
model_kwargs = {