mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-11-21 03:11:40 +08:00
11 lines
360 B
Python
11 lines
360 B
Python
import os.path
|
|
|
|
|
|
def should_hijack_ip2p(checkpoint_info):
|
|
from modules import sd_models_config
|
|
|
|
ckpt_basename = os.path.basename(checkpoint_info.filename).lower()
|
|
cfg_basename = os.path.basename(sd_models_config.find_checkpoint_config_near_filename(checkpoint_info)).lower()
|
|
|
|
return "pix2pix" in ckpt_basename and "pix2pix" not in cfg_basename
|