mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-06-07 21:20:49 +00:00
ee0a0da324
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
14 lines
383 B
Python
14 lines
383 B
Python
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
|