Commit cc8c9b74 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix broken calls to find_checkpoint_config

parent 9beb794e
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -6,7 +6,7 @@ import shutil
import torch
import tqdm

from modules import shared, images, sd_models, sd_vae
from modules import shared, images, sd_models, sd_vae, sd_models_config
from modules.ui_common import plaintext_to_html
import gradio as gr
import safetensors.torch
@@ -37,7 +37,7 @@ def run_pnginfo(image):

def create_config(ckpt_result, config_source, a, b, c):
    def config(x):
        res = sd_models.find_checkpoint_config(x) if x else None
        res = sd_models_config.find_checkpoint_config_near_filename(x) if x else None
        return res if res != shared.sd_default_config else None

    if config_source == 0:
+2 −2
Original line number Diff line number Diff line
@@ -5,9 +5,9 @@ import gc
import time

def should_hijack_ip2p(checkpoint_info):
    from modules import sd_models
    from modules import sd_models_config

    ckpt_basename = os.path.basename(checkpoint_info.filename).lower()
    cfg_basename = os.path.basename(sd_models.find_checkpoint_config(checkpoint_info)).lower()
    cfg_basename = os.path.basename(sd_models_config.find_checkpoint_config_near_filename(checkpoint_info)).lower()

    return "pix2pix" in ckpt_basename and not "pix2pix" in cfg_basename