From cfa883767fba62cd47b300ee6668acb43ef40195 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 24 Oct 2025 12:50:49 +0300 Subject: [PATCH] Update utils.py --- utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils.py b/utils.py index 131b634..5b155c6 100644 --- a/utils.py +++ b/utils.py @@ -507,7 +507,7 @@ def compile_model(transformer, compile_args=None): torch._dynamo.config.force_parameter_static_shapes = compile_args["force_parameter_static_shapes"] try: if hasattr(torch._dynamo.config, 'allow_unspec_int_on_nn_module'): - torch._dynamo.config.allow_unspec_int_on_nn_module = compile_args["allow_unspec_int_on_nn_module"] + torch._dynamo.config.allow_unspec_int_on_nn_module = True except Exception as e: log.warning(f"Could not set allow_unspec_int_on_nn_module: {e}") try: