许可证: cc-by-nc-4.0 数据集:

• asset
• wi_locness
• GEM/wiki_auto_asset_turk
• discofuse
• zaemyung/IteraTeR_plus
• jfleg
• grammarly/coedit 语言:
• en 评估指标:
• sari
• bleu
• accuracy

CoEdIT-xl-composite模型卡

该模型通过在CoEdIT-Composite数据集上微调对应的google/flan-t5-xl模型获得。数据集详情请参阅我们的论文和代码库。

论文标题: CoEdIT: 通过任务特定指令调优实现文本编辑
作者: Vipul Raheja, Dhruv Kumar, Ryan Koo, Dongyeop Kang

模型详情

模型描述

• 支持语言(NLP): 英语
• 基础模型: google/flan-t5-xl

模型来源

• 代码库: https://github.com/vipulraheja/coedit
• 论文: https://arxiv.org/abs/2305.09857

使用方法

我们公开了论文中涉及的所有模型。

功能应用

文本修订任务

给定编辑指令和原始文本，本模型可生成修订后的文本版本。

该模型还能执行复合指令编辑，如下图所示：

调用示例

from transformers import AutoTokenizer, T5ForConditionalGeneration

tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-xl-composite")
model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-xl-composite")
input_text = '修正该句语法错误并简化表达：当我长大后，我开始明白他说的话相当正确。'
input_ids = tokenizer(input_text, return_tensors="pt").input_ids
outputs = model.generate(input_ids, max_length=256)
edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)

相关软件

https://github.com/vipulraheja/coedit

引用

BibTeX格式:

@article{raheja2023coedit,
      title={CoEdIT: 通过任务特定指令调优实现文本编辑}, 
      author={Vipul Raheja and Dhruv Kumar and Ryan Koo and Dongyeop Kang},
      year={2023},
      eprint={2305.09857},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

APA格式:
Raheja, V., Kumar, D., Koo, R., & Kang, D. (2023). CoEdIT: 通过任务特定指令调优实现文本编辑. 预印本论文. /abs/2305.09857