stable-diffusion-webui
10 строк · 360.0 Байт
1import os.path
2
3
4def should_hijack_ip2p(checkpoint_info):
5from modules import sd_models_config
6
7ckpt_basename = os.path.basename(checkpoint_info.filename).lower()
8cfg_basename = os.path.basename(sd_models_config.find_checkpoint_config_near_filename(checkpoint_info)).lower()
9
10return "pix2pix" in ckpt_basename and "pix2pix" not in cfg_basename
11