rust-bert-即用型NLP管道和基于变压器的模型(BERT,DistilBERT,GPT2等)-Rust开发

时间:2024-06-16 12:16:06
【文件属性】:

文件名称:rust-bert-即用型NLP管道和基于变压器的模型(BERT,DistilBERT,GPT2等)-Rust开发

文件大小:460KB

文件格式:ZIP

更新时间:2024-06-16 12:16:06

rust-bert Rust基于Transformer的本机模型实现。 Huggingface的Transformers库的端口,使用tch-rs板条箱和预处理frust-bert Rust基于Transformer的本机模型实现。 Huggingface的Transformers库的端口,使用tch-rs板条箱和防锈剂进行预处理。 支持多线程标记和GPU推理。 该存储库提供了模型基础架构,特定于任务的负责人(请参见下文)和即用型管道。 当前实现了以下模型:DistilBERT BERT RoBERTa GPT GPT2 BART Electra Marian ALBERT Masked LM


网友评论