比如用T5模型,训练数据是inputText-outputText格式,预测时do_sample=False

# 预测代码

from transformers import TFAutoModelForSeq2SeqLM
from transformers import AutoTokenizer


checkpoint_local = "./path/"
tokenizer = AutoTokenizer.from_pretrained(checkpoint_local)
model = TFAutoModelForSeq2SeqLM.from_pretrained(checkpoint_local, 
                                        

点赞(0) 打赏

评论列表 共有 0 条评论

暂无评论

微信公众账号

微信扫一扫加关注

发表
评论
返回
顶部