add create_causal_mask

kylesayrs · kylesayrs · commit 49e32c5f4675 · 2025-06-26T13:18:33.000-04:00
Signed-off-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/src/llmcompressor/args/dataset_arguments.py b/src/llmcompressor/args/dataset_arguments.py
@@ -181,7 +181,7 @@ class DatasetArguments(CustomDatasetArguments):
         },
     )
     tracing_ignore: List[str] = field(
-        default_factory=lambda: ["_update_causal_mask"],
+        default_factory=lambda: ["_update_causal_mask", "create_causal_mask"],
         metadata={
             "help": "List of functions to ignore during tracing, either "
             "{module}.{method_name} or {function_name}"

Original file line number	Diff line number	Diff line change
`@@ -181,7 +181,7 @@ class DatasetArguments(CustomDatasetArguments):`
`181`	`181`	`},`
`182`	`182`	`)`
`183`	`183`	`tracing_ignore: List[str] = field(`
`184`		`- default_factory=lambda: ["_update_causal_mask"],`
	`184`	`+ default_factory=lambda: ["_update_causal_mask", "create_causal_mask"],`
`185`	`185`	`metadata={`
`186`	`186`	`"help": "List of functions to ignore during tracing, either "`
`187`	`187`	`"{module}.{method_name} or {function_name}"`