Add instruct-pix2pix hijack

Allows loading instruct-pix2pix models via same method as inpainting models in sd_models.py and sd_hijack_ip2p.py

Adds ddpm_edit.py necessary for instruct-pix2pix
This commit is contained in:
Kyle
2023-01-25 08:53:23 -05:00
parent d5ce044bcd
commit ee0a0da324
3 changed files with 1483 additions and 1 deletions

13
modules/sd_hijack_ip2p.py Normal file
View File

@@ -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