From 84fed4dc3211a3c10022af0da844a0a8eebfb5c2 Mon Sep 17 00:00:00 2001
From: micsama <1246659083@qq.com>
Date: Wed, 25 Sep 2024 12:01:43 +0800
Subject: [PATCH] fix

---
 paddleslim/nas/ofa/layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddleslim/nas/ofa/layers.py b/paddleslim/nas/ofa/layers.py
index 892b52591..51c090a81 100644
--- a/paddleslim/nas/ofa/layers.py
+++ b/paddleslim/nas/ofa/layers.py
@@ -1298,7 +1298,7 @@ def forward(self, input):
         self.cur_config = {'prune_dim': feature_dim}
 
         if paddle.in_dynamic_mode():
-            out, _, _ = paddle._C_ops.layer_norm(
+            out = paddle._C_ops.layer_norm(
                 input, weight, bias, self._epsilon, begin_norm_axis, False)
         else:
             paddle.common_ops_import.check_variable_and_dtype(