From 526bb5e4c4a80b17d0e7fa265152a680d7e5e4de Mon Sep 17 00:00:00 2001 From: facok <128763816+facok@users.noreply.github.com> Date: Wed, 8 Jan 2025 14:03:04 +0800 Subject: [PATCH] fix fp8 --- nodes.py | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/nodes.py b/nodes.py index 463f2f9..3aff080 100644 --- a/nodes.py +++ b/nodes.py @@ -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" -} \ No newline at end of file +}