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

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-27 19:46:55 -08:00
parent 2074e13253
commit 4443d636c6
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,6 +9,7 @@ from modules.shared_cmd_options import cmd_opts
from modules.paths import models_path from modules.paths import models_path
from modules.modelloader import load_file_from_url from modules.modelloader import load_file_from_url
from ldm_patched.modules.controlnet import load_controlnet from ldm_patched.modules.controlnet import load_controlnet
from modules_forge.controlnet import apply_controlnet_advanced
class ControlNetExampleForge(scripts.Script): class ControlNetExampleForge(scripts.Script):
@@ -77,7 +78,9 @@ class ControlNetExampleForge(scripts.Script):
unet = p.sd_model.forge_objects.unet unet = p.sd_model.forge_objects.unet
# unet = opFreeU_V2.patch(unet, freeu_b1, freeu_b2, freeu_s1, freeu_s2)[0] unet = apply_controlnet_advanced(unet=unet, controlnet=self.model, image=canny_image,
strength=1.0, start_percent=0.0, end_percent=1.0,
advanced_weighting=None)
p.sd_model.forge_objects.unet = unet p.sd_model.forge_objects.unet = unet

2
modules_forge/controlnet.py Normal file
View File

@@ -0,0 +1,2 @@
def apply_controlnet_advanced(unet, controlnet, image, strength, start_percent, end_percent, advanced_weighting=None):
return unet