CoEdIT: Text Editing by Task-Specific Instruction Tuning
Paper
•
2305.09857
•
Published
•
7
YAML Metadata
Warning:
The pipeline tag "text2text-generation" is not in the official list: text-classification, token-classification, table-question-answering, question-answering, zero-shot-classification, translation, summarization, feature-extraction, text-generation, fill-mask, sentence-similarity, text-to-speech, text-to-audio, automatic-speech-recognition, audio-to-audio, audio-classification, audio-text-to-text, voice-activity-detection, depth-estimation, image-classification, object-detection, image-segmentation, text-to-image, image-to-text, image-to-image, image-to-video, unconditional-image-generation, video-classification, reinforcement-learning, robotics, tabular-classification, tabular-regression, tabular-to-text, table-to-text, multiple-choice, text-ranking, text-retrieval, time-series-forecasting, text-to-video, image-text-to-text, image-text-to-image, image-text-to-video, visual-question-answering, document-question-answering, zero-shot-image-classification, graph-ml, mask-generation, zero-shot-object-detection, text-to-3d, image-to-3d, image-feature-extraction, video-text-to-text, keypoint-detection, visual-document-retrieval, any-to-any, video-to-video, other
CoEdIT Large (ONNX)
ONNX quantized version of Grammarly's CoEdIT-Large model for browser-based text editing and grammatical error correction with Transformers.js.
Original Model
- Source: grammarly/coedit-large
- Paper: CoEdIT: Text Editing by Task-Specific Instruction Tuning (EMNLP 2023)
- Architecture: Flan-T5-Large (encoder-decoder)
- Parameters: ~770M
Conversion Details
- Format: ONNX
- Quantization: INT8 (dynamic quantization)
- Size: ~780MB
- Converted by: Optimum CLI
How It Works
CoEdIT is a text-to-text model fine-tuned for various editing tasks. For grammar correction, use the prompt format:
Fix grammatical errors: <input text>
The model generates the corrected text directly.
Usage with 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."
Supported Tasks
CoEdIT supports multiple editing tasks via different prompts:
Fix grammatical errors: ...- Grammar correctionParaphrase: ...- ParaphrasingSimplify: ...- Text simplificationMake this more formal: ...- Formality transferMake this more coherent: ...- Coherence improvement
Performance
More flexible than GECToR (can rewrite sentences), but slower due to autoregressive generation.
License
Apache 2.0 (following original model license)
- Downloads last month
- 8