🚀 燃灯-Deltalm-362M-中-英
本模型是一个中英翻译模型,使用封神框架基于 Detalm base 进行微调,利用了搜集的中英数据集(共三千万条)以及 iwslt 的中英平行数据(二十万条),在中 -> 英方向的翻译任务上表现出色。
🚀 快速开始
✨ 主要特性
使用封神框架基于 Detalm base 进行 finetune,利用丰富的中英数据集,得到中 -> 英方向的翻译模型。
📦 安装指南
使用前,你需要提前从 Fengshenbang-LM 的 GitHub 仓库下载 modeling_deltalm.py
,或者使用以下链接下载:modeling_deltalm.py。强烈建议你克隆 Fengshenbang-LM 仓库:
git clone https://github.com/IDEA-CCNL/Fengshenbang-LM
cd Fengshenbang-LM/fengshen/models/deltalm/
💻 使用示例
基础用法
from modeling_deltalm import DeltalmForConditionalGeneration
from transformers import AutoTokenizer
model = DeltalmForConditionalGeneration.from_pretrained("IDEA-CCNL/Randeng-Deltalm-362M-Zh-En")
tokenizer = AutoTokenizer.from_pretrained("microsoft/infoxlm-base")
text = "尤其在夏天,如果你决定徒步穿越雨林,就需要小心蚊子。"
inputs = tokenizer(text, max_length=512, return_tensors="pt")
generate_ids = model.generate(inputs["input_ids"], max_length=512)
tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
📚 详细文档
模型分类
属性 |
详情 |
需求 |
通用 |
任务 |
自然语言转换 |
系列 |
燃灯 |
模型 |
Deltalm |
参数 |
362M |
额外 |
翻译任务(中-英) |
模型信息
参考论文:DeltaLM: Encoder-Decoder Pre-training for Language Generation and Translation by Augmenting Pretrained Multilingual Encoders
下游效果
数据集 |
BLEU |
florse101-zh-en |
26.47 |
📄 许可证
本项目未明确提及许可证相关信息。
🔗 引用
如果您在您的工作中使用了我们的模型,可以引用我们的论文:
@article{fengshenbang,
author = {Jiaxing Zhang and Ruyi Gan and Junjie Wang and Yuxiang Zhang and Lin Zhang and Ping Yang and Xinyu Gao and Ziwei Wu and Xiaoqun Dong and Junqing He and Jianheng Zhuo and Qi Yang and Yongfeng Huang and Xiayu Li and Yanghan Wu and Junyu Lu and Xinyu Zhu and Weifeng Chen and Ting Han and Kunhao Pan and Rui Wang and Hao Wang and Xiaojun Wu and Zhongshen Zeng and Chongpei Chen},
title = {Fengshenbang 1.0: Being the Foundation of Chinese Cognitive Intelligence},
journal = {CoRR},
volume = {abs/2209.02970},
year = {2022}
}
也可以引用我们的网站:
@misc{Fengshenbang-LM,
title={Fengshenbang-LM},
author={IDEA-CCNL},
year={2021},
howpublished={\url{https://github.com/IDEA-CCNL/Fengshenbang-LM}},
}