CoEdIT Large (ONNX)

Grammarly CoEdIT-Large 模型的 ONNX 量化版本，用于基于浏览器的文本编辑和语法错误纠正，使用 Transformers.js。

原始模型

• 来源: grammarly/coedit-large
• 论文: CoEdIT: Text Editing by Task-Specific Instruction Tuning (EMNLP 2023)
• 架构: Flan-T5-Large (编码器-解码器)
• 参数量: ~770M

转换详情

• 格式: ONNX
• 量化: INT8 (动态量化)
• 大小: ~780MB
• 转换工具: Optimum CLI

工作原理

CoEdIT 是一个针对各种编辑任务进行微调的文本到文本模型。对于语法纠正，请使用以下提示格式:

Fix grammatical errors: <input text>

模型将直接生成纠正后的文本。

使用 Transformers.js

import { pipeline } from '@huggingface/transformers';

const corrector = await pipeline(
  'text2text-generation',
  'YOUR_USERNAME/coedit-large',
  { dtype: 'q8' }
);

const result = await corrector('Fix grammatical errors: He go to school yesterday.');
console.log(result[0].generated_text);
// \"He went to school yesterday.\"

支持的任务

CoEdIT 通过不同的提示支持多种编辑任务:

• Fix grammatical errors: ... - 语法纠正
• Paraphrase: ... - 改写
• Simplify: ... - 文本简化
• Make this more formal: ... - 正式化转换
• Make this more coherent: ... - 连贯性改进

性能

比 GECToR 更灵活 (可以重写句子)，但由于自回归生成，速度较慢。

许可证

Apache 2.0 (遵循原始模型许可证)