PaddleNLP icon indicating copy to clipboard operation
PaddleNLP copied to clipboard

Published 20 hours ago • verified publisher icon PaddlePaddle

fix the bug that device_id invalid in fp16 mode

Open linjieccc opened this issue 2 years ago • 0 comments

PR types

Bug fixes

PR changes

APIs

Description

  • 修复Taskflow UIE在半精度FP16推理时device_id失效的问题 #2952
  • uie_predictor新增device_id用于指定GPU ID

linjieccc avatar Aug 04 '22 03:08 linjieccc