HLK-ViT5

HLK-ViT5 là mô hình tóm tắt văn bản tiếng Việt được fine-tune từ VietAI/vit5-base.

Base model

• VietAI/vit5-base

Task

• Vietnamese text summarization
• Text-to-text generation

Framework

• PyTorch
• Hugging Face Transformers

Usage

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

model_id = "tmanh217/hlk-vit5"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSeq2SeqLM.from_pretrained(model_id)