How to use transformers pipeline with multi-gpu?

后端 未结 0 1522
一整个雨季
一整个雨季 2021-01-21 09:15
ner_model = pipeline(\'ner\', model=model, tokenizer=tokenizer, device=0, grouped_entities=True)

the device indicated pipeline

相关标签:
回答
  • 消灭零回复
提交回复
热议问题