T5-Pegasus-Pytorch是一款结合了Google T5与Salesforce Pegasus模型特点,并基于Pytorch框架开发的语言处理工具包,适用于多种自然语言理解及生成任务。
t5-pegasus 的 pytorch 版本由追一科技开源。你可以下载 base 版本。
要使用该版本,请确保你的环境配置为:pytorch 1.7.0 + transformers 4.3.3
导入所需的库:
```python
from tokenizer import T5PegasusTokenizer
from transformers.models.mt5.modeling_mt5 import MT5ForConditionalGeneration
```
加载模型和分词器的路径(假设你已经将它们下载到了本地):
```python
model_path = ./
model = MT5ForConditionalGeneration.from_pretrained(model_path)
tokenizer = T5PegasusTokenizer.from_pretrained(model_path)
```
示例文本为 蓝。