diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-25 15:33:15 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-25 15:33:15 +0000 |
| commit | 11485659dca08ef967f3e5462382f91504195ef0 (patch) | |
| tree | aa93f5921b150effba36510af1f46668346f8b50 /modules/sd_hijack_ip2p.py | |
| parent | d5ce044bcdd9593751d7e0ba28087247ff0b7926 (diff) | |
| parent | bd9b55ee908c43fb1b654b3a3a1320545023ce1c (diff) | |
| download | stable-diffusion-webui-gfx803-11485659dca08ef967f3e5462382f91504195ef0.tar.gz stable-diffusion-webui-gfx803-11485659dca08ef967f3e5462382f91504195ef0.tar.bz2 stable-diffusion-webui-gfx803-11485659dca08ef967f3e5462382f91504195ef0.zip | |
Merge pull request #7195 from Klace/instruct-pix2pix_model_load
Add instruct-pix2pix hijack
Diffstat (limited to 'modules/sd_hijack_ip2p.py')
| -rw-r--r-- | modules/sd_hijack_ip2p.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/modules/sd_hijack_ip2p.py b/modules/sd_hijack_ip2p.py new file mode 100644 index 00000000..635f015f --- /dev/null +++ b/modules/sd_hijack_ip2p.py @@ -0,0 +1,13 @@ +import collections +import os.path +import sys +import gc +import time + +def should_hijack_ip2p(checkpoint_info): + from modules import sd_models + + ckpt_basename = os.path.basename(checkpoint_info.filename).lower() + cfg_basename = os.path.basename(sd_models.find_checkpoint_config(checkpoint_info)).lower() + + return "pix2pix" in ckpt_basename and not "pix2pix" in cfg_basename |