mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-25 12:39:25 +01:00
[to #42322933] update preprocessor for deberta v2
相关改动已经在 model repo中进行了改动 Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11714125 * update preprocessor for deberta
This commit is contained in:
committed by
wenmeng.zwm
parent
3e4ee5958a
commit
fb9c41092a
@@ -75,11 +75,6 @@ class NLPTokenizer:
|
||||
# fast version lead to parallel inference failed
|
||||
model_type = self.model_type
|
||||
model_dir = self.model_dir
|
||||
if model_type == Models.deberta_v2:
|
||||
from modelscope.models.nlp.deberta_v2 import DebertaV2Tokenizer, DebertaV2TokenizerFast
|
||||
tokenizer = DebertaV2TokenizerFast if self.use_fast else DebertaV2Tokenizer
|
||||
return tokenizer.from_pretrained(
|
||||
model_dir) if model_dir is not None else tokenizer()
|
||||
|
||||
if model_type in (Models.structbert, Models.gpt3, Models.palm,
|
||||
Models.plug, Models.megatron_bert,
|
||||
|
||||
Reference in New Issue
Block a user