jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Move controlnet code to comfy/controlnet.py

Browse Source
This commit is contained in:
comfyanonymous
2023-08-25 17:25:39 -04:00
parent 15a7716fa6
commit c77f02e1c6
4 changed files with 533 additions and 527 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
View File

@@ -22,6 +22,7 @@ import comfy.samplers
import comfy.sample
import comfy.sd
import comfy.utils
import comfy.controlnet
import comfy.clip_vision
@@ -569,7 +570,7 @@ class ControlNetLoader:
def load_controlnet(self, control_net_name):
controlnet_path = folder_paths.get_full_path("controlnet", control_net_name)
controlnet = comfy.sd.load_controlnet(controlnet_path)
controlnet = comfy.controlnet.load_controlnet(controlnet_path)
return (controlnet,)
class DiffControlNetLoader:
@@ -585,7 +586,7 @@ class DiffControlNetLoader:
def load_controlnet(self, model, control_net_name):
controlnet_path = folder_paths.get_full_path("controlnet", control_net_name)
controlnet = comfy.sd.load_controlnet(controlnet_path, model)
controlnet = comfy.controlnet.load_controlnet(controlnet_path, model)
return (controlnet,)