From 2951008ddee5f781264376b178989108926ebc46 Mon Sep 17 00:00:00 2001 From: yyyyrf Date: Fri, 27 Dec 2024 15:32:55 +0800 Subject: [PATCH] ffn forward x shape --- .../ptq/ptq/wrappers/mindformers/linear_smooth_wrappers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore_gs/ptq/ptq/wrappers/mindformers/linear_smooth_wrappers.py b/mindspore_gs/ptq/ptq/wrappers/mindformers/linear_smooth_wrappers.py index 22067c15..85e6a499 100644 --- a/mindspore_gs/ptq/ptq/wrappers/mindformers/linear_smooth_wrappers.py +++ b/mindspore_gs/ptq/ptq/wrappers/mindformers/linear_smooth_wrappers.py @@ -406,7 +406,7 @@ class AWQSmoothLinearCell(AWQLinearCell): self.forward_module = self.decoder.feed_forward else: self.forward_module = self._layer - return self.forward_module(self.cat_samples) + return self.forward_module(self.cat_samples.expand_dims(0)) def smooth(self): """smooth""" -- Gitee