fix broken calls to find_checkpoint_config

2 files changed, 4 insertions, 4 deletions

diff --git a/modules/extras.py b/modules/extras.py
index 36123aa5..4f842be9 100644
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -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:

diff --git a/modules/sd_hijack_ip2p.py b/modules/sd_hijack_ip2p.py
index 635f015f..3c727d3b 100644
--- a/modules/sd_hijack_ip2p.py
+++ b/modules/sd_hijack_ip2p.py
@@ -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