search

Albert-Ma / PROP

WSDM'2021, PROP and SIGIR'2021,B-PROP
Apache License 2.0
110 stars 18 forks source link

发布的模型文件可以直接使用huggingface transformers来Load吗？ #2

Closed zhengyima closed 3 years ago

zhengyima commented 3 years ago

谢谢

Albert-Ma commented 3 years ago

可以的，你把模型文件名字命名成pytorch_model.bin，在下载config.json 和 vocab.txt放在同一个目录下，load模型的时候调用from_pretrained(MODEL_PATH)就可以了

具体可以看这里link, huggingface transformers 对一些模型的名字有一些预定义

zhengyima commented 3 years ago

好的，已经实验成功了。