pytorch-transformers

1热度

5回答

[huggingface/transformers]mxmax/Chinese_Chat_T5_Base模型怎么用torch.jit.trace追踪

我使用下面的代码进行模型转换 tokenizer = AutoTokenizer.from_pretrained('./outputs/model_files/') model = AutoModelForSeq2SeqLM.from_pretrained('./outputs/model_files/') device = torch.device("cpu") model.to(device)

Deep learning Machine learning pytorch-transformers 2024-01-23

---