diff --git a/src/llmcompressor/pipelines/sequential/helpers.py b/src/llmcompressor/pipelines/sequential/helpers.py index f80cb65a7..51e7c3a74 100644 --- a/src/llmcompressor/pipelines/sequential/helpers.py +++ b/src/llmcompressor/pipelines/sequential/helpers.py @@ -74,9 +74,7 @@ def forward(self, *args, **kwargs) -> Dict[str, Any]: except Exception as exception: 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" + f"```\n{add_line_numbers(self._code.src)}\n```" ) from exception return outputs