diff --git a/src/llmcompressor/pipelines/sequential/helpers.py b/src/llmcompressor/pipelines/sequential/helpers.py index d2e4988ee..6f19b8ade 100644 --- a/src/llmcompressor/pipelines/sequential/helpers.py +++ b/src/llmcompressor/pipelines/sequential/helpers.py @@ -74,8 +74,6 @@ def forward(self, *args, **kwargs) -> Dict[str, Any]: raise RuntimeError( "Raised an exception during execution of the following code:\n" f"```\n{add_line_numbers(self._code.src)}\n```\n" - "This is likely due to a violation of shape assumptions made when " - "tracing" ) from exception return outputs