mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-13 21:16:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

ControlNetApplySD3 node can now be used to use SD3 controlnets.

Browse Source
This commit is contained in:
comfyanonymous
2024-06-26 16:14:47 -04:00
parent f8f7568d03
commit 264caca20e
4 changed files with 41 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/cldm/mmdit.py
View File

@@ -1,7 +1,6 @@
import torch
from typing import Dict, Optional
import comfy.ldm.modules.diffusionmodules.mmdit
import comfy.latent_formats
class ControlNet(comfy.ldm.modules.diffusionmodules.mmdit.MMDiT):
def __init__(
@@ -30,8 +29,6 @@ class ControlNet(comfy.ldm.modules.diffusionmodules.mmdit.MMDiT):
operations=operations
)
self.latent_format = comfy.latent_formats.SD3()
def forward(
self,
x: torch.Tensor,
@@ -42,10 +39,8 @@ class ControlNet(comfy.ldm.modules.diffusionmodules.mmdit.MMDiT):
) -> torch.Tensor:
#weird sd3 controlnet specific stuff
hint = hint * self.latent_format.scale_factor # self.latent_format.process_in(hint)
y = torch.zeros_like(y)
if self.context_processor is not None:
context = self.context_processor(context)