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

Added EasyCache support to hunyuan3d

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-08-21 23:38:14 -07:00
parent ca5ad630cd
commit d60e7a07cb
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/ldm/hunyuan3d/model.py
View File

@ -7,6 +7,7 @@ from comfy.ldm.flux.layers import (
SingleStreamBlock, SingleStreamBlock,
timestep_embedding, timestep_embedding,
) )
import comfy.patcher_extension
class Hunyuan3Dv2(nn.Module): class Hunyuan3Dv2(nn.Module):
@ -67,6 +68,13 @@ class Hunyuan3Dv2(nn.Module):
self.final_layer = LastLayer(hidden_size, 1, in_channels, dtype=dtype, device=device, operations=operations) self.final_layer = LastLayer(hidden_size, 1, in_channels, dtype=dtype, device=device, operations=operations)
def forward(self, x, timestep, context, guidance=None, transformer_options={}, **kwargs): def forward(self, x, timestep, context, guidance=None, transformer_options={}, **kwargs):
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, timestep, context, guidance, transformer_options, **kwargs)
def _forward(self, x, timestep, context, guidance=None, transformer_options={}, **kwargs):
x = x.movedim(-1, -2) x = x.movedim(-1, -2)
timestep = 1.0 - timestep timestep = 1.0 - timestep
txt = context txt = context