Skip to content

Commit 2150ae3

Browse files
authored
fix use paddle.incubate.jit.inference(model) errors (#9016)
* fix use paddle.incubate.jit.inference(model) errors * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function * fix jit_inference function
1 parent ce0c0e7 commit 2150ae3

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

paddlenlp/transformers/utils.py

+12
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,18 @@ def adapt_stale_fwd_patch(self, name, value):
167167
"StaticFunction"
168168
):
169169
return value
170+
171+
# NOTE(changwenbin & zhoukangkang):
172+
# When use model = paddle.incubate.jit.inference(model), it reportes errors, we fix it here.
173+
# is_inference_mode API is only avaliable in PaddlePaddle develop,so we add a try except.
174+
try:
175+
from paddle.incubate.jit import is_inference_mode
176+
177+
if is_inference_mode(value):
178+
return value
179+
except:
180+
pass
181+
170182
if hasattr(inspect, "getfullargspec"):
171183
(
172184
patch_spec_args,

0 commit comments

Comments
 (0)