fix broken calls to find_checkpoint_config

This commit is contained in:
AUTOMATIC
2023-01-27 22:43:08 +03:00
parent 9beb794e0b
commit cc8c9b7474
2 changed files with 4 additions and 4 deletions

View File

@@ -5,9 +5,9 @@ import gc
import time
def should_hijack_ip2p(checkpoint_info):
from modules import sd_models
from modules import sd_models_config
ckpt_basename = os.path.basename(checkpoint_info.filename).lower()
cfg_basename = os.path.basename(sd_models.find_checkpoint_config(checkpoint_info)).lower()
cfg_basename = os.path.basename(sd_models_config.find_checkpoint_config_near_filename(checkpoint_info)).lower()
return "pix2pix" in ckpt_basename and not "pix2pix" in cfg_basename