From 82c76914a23f4f5671997d91c45529ed1160ba2e Mon Sep 17 00:00:00 2001 From: digital-garbage <133010426+digital-garbage@users.noreply.github.com> Date: Mon, 18 May 2026 19:45:52 +0300 Subject: [PATCH] Fix model_function_wrapper: unpack args dict as positional (x, t, **c) apply_model takes (x, t, **c) not (args_dict). Both wrapper call sites were passing the whole dict as x, leaving t unfilled. Co-Authored-By: Claude Sonnet 4.6 --- ltx_enhancements.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ltx_enhancements.py b/ltx_enhancements.py index e996b02e..8f2fd55a 100644 --- a/ltx_enhancements.py +++ b/ltx_enhancements.py @@ -341,7 +341,7 @@ def build_enhancements(model, rating_profile, temporal_style, refinement_key, re pass if _old is not None: return _old(apply_fn, args) - return apply_fn(args) + return apply_fn(args["input"], args["timestep"], **args.get("c", {})) model.model_options["model_function_wrapper"] = _temporal_wrapper @@ -358,7 +358,7 @@ def build_enhancements(model, rating_profile, temporal_style, refinement_key, re if _ew is not None: result = _ew(apply_fn, args) else: - result = apply_fn(args) + result = apply_fn(args["input"], args["timestep"], **args.get("c", {})) # Opportunistically save the capture buffer when it has anchor maps. # We do this every call so the last (cleanest) capture wins.