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
This commit is contained in:
AIbin
2024-09-06 17:35:27 +08:00
committed by GitHub
parent ce0c0e73c5
commit 2150ae3539

View File

@@ -167,6 +167,18 @@ def adapt_stale_fwd_patch(self, name, value):
"StaticFunction"
):
return value
# NOTE(changwenbin & zhoukangkang):
# When use model = paddle.incubate.jit.inference(model), it reportes errors, we fix it here.
# is_inference_mode API is only avaliable in PaddlePaddle develop,so we add a try except.
try:
from paddle.incubate.jit import is_inference_mode
if is_inference_mode(value):
return value
except:
pass
if hasattr(inspect, "getfullargspec"):
(
patch_spec_args,