Skip to content

Commit

Permalink
Revert "Fit pir flag in predictor (#8048)" (#8065)
Browse files Browse the repository at this point in the history
This reverts commit 5df6e7d.
  • Loading branch information
zjjlivein authored Mar 6, 2024
1 parent 67f31c6 commit fb0ddb5
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions llm/predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,6 @@
import paddle
import paddle.distributed.fleet.base.topology as tp
import paddle.incubate.multiprocessing as mp
from paddle.base.framework import in_cinn_mode, in_pir_executor_mode
from paddle.distributed import fleet
from utils import (
dybatch_preprocess,
Expand Down Expand Up @@ -361,10 +360,6 @@ def __init__(self, config: PredictorArgument, tokenizer: PretrainedTokenizer = N
inference_config.disable_gpu()
inference_config.disable_glog_info()
inference_config.enable_new_executor()
if in_pir_executor_mode():
inference_config.enable_new_ir()
if in_cinn_mode():
inference_config.enable_cinn()

with static_mode_guard():
self.predictor = paddle.inference.create_predictor(inference_config)
Expand Down

0 comments on commit fb0ddb5

Please sign in to comment.