xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 00:16:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add EasyCache support to hidream

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-08-21 22:29:41 -07:00
parent 4bc3e217d6
commit d93c5a9830
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
comfy/ldm/hidream/model.py
View File

@ -13,6 +13,7 @@ from comfy.ldm.flux.layers import LastLayer
from comfy.ldm.modules.attention import optimized_attention from comfy.ldm.modules.attention import optimized_attention
import comfy.model_management import comfy.model_management
import comfy.patcher_extension
import comfy.ldm.common_dit import comfy.ldm.common_dit
@ -692,7 +693,23 @@ class HiDreamImageTransformer2DModel(nn.Module):
raise NotImplementedError raise NotImplementedError
return x, x_masks, img_sizes return x, x_masks, img_sizes
def forward( def forward(self,
x: torch.Tensor,
t: torch.Tensor,
y: Optional[torch.Tensor] = None,
context: Optional[torch.Tensor] = None,
encoder_hidden_states_llama3=None,
image_cond=None,
control = None,
transformer_options = {},
):
return comfy.patcher_extension.WrapperExecutor.new_class_executor(
self._forward,
self,
comfy.patcher_extension.get_all_wrappers(comfy.patcher_extension.WrappersMP.DIFFUSION_MODEL, transformer_options)
).execute(x, t, y, context, encoder_hidden_states_llama3, image_cond, control, transformer_options)
def _forward(
self, self,
x: torch.Tensor, x: torch.Tensor,
t: torch.Tensor, t: torch.Tensor,