pick correct config file for sdxl models

This commit is contained in:
Lincoln Stein
2023-08-07 15:19:49 -04:00
parent ae17d01e1d
commit af044007d5
2 changed files with 7 additions and 3 deletions

View File

@ -80,8 +80,10 @@ class StableDiffusionXLModel(DiffusersModel):
raise Exception("Unkown stable diffusion 2.* model format")
if ckpt_config_path is None:
# TO DO: implement picking
pass
# avoid circular import
from .stable_diffusion import _select_ckpt_config
ckpt_config_path = _select_ckpt_config(kwargs.get("model_base", BaseModelType.StableDiffusionXL), variant)
return cls.create_config(
path=path,