find configs for models at runtime rather than when starting

This commit is contained in:
AUTOMATIC
2023-01-04 12:47:42 +03:00
parent 02d7abf514
commit 8d8a05a3bb
2 changed files with 22 additions and 14 deletions

View File

@@ -97,8 +97,11 @@ def p_sample_plms(self, x, c, t, index, repeat_noise=False, use_original_steps=F
def should_hijack_inpainting(checkpoint_info):
from modules import sd_models
ckpt_basename = os.path.basename(checkpoint_info.filename).lower()
cfg_basename = os.path.basename(checkpoint_info.config).lower()
cfg_basename = os.path.basename(sd_models.find_checkpoint_config(checkpoint_info)).lower()
return "inpainting" in ckpt_basename and not "inpainting" in cfg_basename