Views
No views yet
IDEA-CCNL/Randeng-BART-139M-SUMMARY and is used by the project backend for text-to-text generation.BartForConditionalGeneration 权重,可直接通过 transformers 加载。项目中主要用于现代汉语到文言风格的转换。IDEA-CCNL/Randeng-BART-139M-SUMMARYBartForConditionalGenerationnum_beams=41from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model_id = "liuyanliang/randeng-bart-modern-to-classical-100k-bs16"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
7model.eval()
8
9text = "今文翻古:学习后按时温习,不也很快乐吗"
10inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)
11outputs = model.generate(**inputs, max_new_tokens=128, num_beams=4)
12print(tokenizer.decode(outputs[0], skip_special_tokens=True))1export CCNLP_SEQ2SEQ_MODEL=liuyanliang/randeng-bart-modern-to-classical-100k-bs16
2
3PYTHONPATH=src uvicorn ccnlp.api_server:app --host 127.0.0.1 --port 80001今文翻古:<modern Chinese input>
2古文翻今:<Classical Chinese input>model.safetensors: fine-tuned BART model weightsconfig.json: model configurationgeneration_config.json: generation defaultstransformers