lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 18:51:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 07:47:56 -08:00
parent bd334d3aff
commit 336eac060f
2 changed files with 27 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py
View File

@@ -78,17 +78,6 @@ class ControlNetExampleForge(scripts.Script):
input_image = cv2.resize(input_image, (width, height))
canny_image = cv2.cvtColor(cv2.Canny(input_image, 100, 200), cv2.COLOR_GRAY2RGB)
from modules_forge.ops import capture_model
from modules_forge.shared import shared_preprocessors
with capture_model() as captured_model:
canny_image = shared_preprocessors['normalbae'](input_image, 512)
captured_model.cpu()
from ldm_patched.modules import model_management
model_management.soft_empty_cache()
a = 0
# # Or you can get a list of preprocessors in this way
# from modules_forge.shared import shared_preprocessors
# canny_preprocessor = shared_preprocessors['canny']