Skip to content

Commit c8d7323

Browse files
committed
update ignore defaults in debug script
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 94e73b0 commit c8d7323

File tree

1 file changed

+2
-2
lines changed
  • src/llmcompressor/transformers/tracing

1 file changed

+2
-2
lines changed

src/llmcompressor/transformers/tracing/debug.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def parse_args():
2222
parser.add_argument("--model_id", type=str, required=True, help="The stub of the model to load") # noqa: E501
2323
parser.add_argument("--model_class", type=str, required=True, help="The class name of the model") # noqa: E501
2424
parser.add_argument("--sequential_targets", type=str, nargs="*", default=None, metavar="TARGET", help="List of targets for sequential tracing") # noqa: E501
25-
parser.add_argument("--ignore", type=str, nargs="*", default=["_update_causal_mask"], metavar="PATTERN", help="List of patterns to ignore during tracing") # noqa: E501
25+
parser.add_argument("--ignore", type=str, nargs="*", default=DatasetArguments().tracing_ignore, metavar="PATTERN", help="List of patterns to ignore during tracing") # noqa: E501
2626
parser.add_argument("--modality", type=str, default="text", help="Modality of calibration dataset, defaults to text") # noqa: E501
2727
parser.add_argument("--trust_remote_code", type=bool, default=False, help="Whether to trust model remote code") # noqa: E501
2828
parser.add_argument("--skip_weights", type=bool, default=True, help="Whether to load the model with dummy weights") # noqa: E501
@@ -34,7 +34,7 @@ def trace(
3434
model_id: str,
3535
model_class: Type[PreTrainedModel],
3636
sequential_targets: Optional[Union[List[str], str]] = None,
37-
ignore: Union[List[str], str] = ["_update_causal_mask"],
37+
ignore: Union[List[str], str] = DatasetArguments().tracing_ignore,
3838
modality: str = "text",
3939
trust_remote_code: bool = True,
4040
skip_weights: bool = True,

0 commit comments

Comments
 (0)