Skip to content

Commit 0160aa3

Browse files
authored
modify npu predict bug (#2454)
1 parent e35f2f1 commit 0160aa3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddlex/inference/components/paddle_predictor/predictor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,8 @@ def _create(self):
181181
config.enable_new_ir(self.option.enable_new_ir)
182182
if hasattr(config, "enable_new_executor"):
183183
config.enable_new_executor()
184-
config.set_optimization_level(3)
184+
if self.option.device in ("cpu", "gpu"):
185+
config.set_optimization_level(3)
185186

186187
for del_p in self.option.delete_pass:
187188
config.delete_pass(del_p)

0 commit comments

Comments
 (0)