Update utils.py

2025-10-24 12:50:49 +03:00
parent 88a60d71ab
commit cfa883767f
1 changed files with 1 additions and 1 deletions
@@ -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: