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 22067c157bb7b02d79728ddba1a754911d3b57c0..85e6a499fc95b5841e4adce883e782aab1febb70 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"""