Skip to content
This repository has been archived by the owner on Dec 10, 2023. It is now read-only.

Commit

Permalink
Fix import
Browse files Browse the repository at this point in the history
  • Loading branch information
OedoSoldier committed May 22, 2023
1 parent a112a44 commit 4de9494
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions scripts/enhanced_img2img.py
Expand Up @@ -22,9 +22,6 @@
if cmd_opts.deepdanbooru:
import modules.deepbooru as deepbooru

import importlib
external_code = importlib.import_module('extensions.sd-webui-controlnet.scripts.external_code', 'external_code')

import re

re_findidx = re.compile(
Expand Down Expand Up @@ -466,6 +463,9 @@ def run(
state.job_count *= len(images)

def set_reference(p, idx, enabled=False):
import importlib
external_code = importlib.import_module('extensions.sd-webui-controlnet.scripts.external_code', 'external_code')

cn_units = external_code.get_all_units_in_processing(p)
cn_units[idx].enabled = enabled
external_code.update_cn_script_in_processing(p, cn_units)
Expand Down

0 comments on commit 4de9494

Please sign in to comment.