Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "161449d51ae24f9b0f4935e8574bbe55c59fe0a4"
Unverified Commit dc20feb1 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #40 from LDOUBLEV/fixocr 

solve predicts diff between cpu and gpu running
parents e807901b 7259370c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tools/infer/utility.py
View file @ dc20feb1
...@@ -91,6 +91,7 @@ def create_predictor(args, mode): ...@@ -91,6 +91,7 @@ def create_predictor(args, mode):
config.disable_glog_info() config.disable_glog_info()
# use zero copy # use zero copy
config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
config.switch_use_feed_fetch_ops(False) config.switch_use_feed_fetch_ops(False)
predictor = create_paddle_predictor(config) predictor = create_paddle_predictor(config)
input_names = predictor.get_input_names() input_names = predictor.get_input_names()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment