fix load bugs (#58159)

PaddlePaddle · Oct 18, 2023 · 2724c80 · 2724c80
1 parent 6f48047
commit 2724c80
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/paddle/jit/translated_layer.py b/python/paddle/jit/translated_layer.py
@@ -512,6 +512,11 @@ def _preprocess(self, program_desc):
 
     @switch_to_static_graph
     def _append_scale_to_output(self, program):
+        # 0. scale don't support bool output, we skip append scale for it
+        for out_desc in self._output_descs:
+            if out_desc.dtype() == core.VarDesc.VarType.BOOL:
+                return
+
         # 1. append scale & save var
         scale_output_vars = []
         with framework.program_guard(program):