This commit is contained in:
facok
2025-01-08 14:03:04 +08:00
committed by GitHub
parent d4c446ceb7
commit 526bb5e4c4
+1 -12
View File
@@ -72,17 +72,6 @@ class TeaCacheHunyuanVideoSampler:
# Get input dimensions
B, C, T, H, W = x.shape
# Ensure consistent input dtypes
try:
x = x.to(dtype=transformer.dtype)
timestep = timestep.to(dtype=transformer.dtype)
if context is not None:
context = context.to(dtype=transformer.dtype)
if guidance is not None:
guidance = guidance.to(dtype=transformer.dtype)
except Exception as e:
raise RuntimeError(f"Failed to convert input dtypes: {str(e)}")
# Prepare modulation vectors
try:
# HunyuanVideo uses timestep_embedding for time step encoding
@@ -272,4 +261,4 @@ NODE_CLASS_MAPPINGS = {
NODE_DISPLAY_NAME_MAPPINGS = {
"TeaCacheHunyuanVideoSampler": "TeaCache HunyuanVideo Sampler"
}
}