ghosthamlet / gpt2-ml-torch

Pytorch model for https://github.com/imcaspar/gpt2-ml
Apache License 2.0
79 stars 16 forks source link

方便给一个google drive的下载链接么? #17

Closed henryxiao1997 closed 3 years ago

henryxiao1997 commented 3 years ago

如题,谢谢!~

ghosthamlet commented 3 years ago

不好意思啊,我没有google帐号,你可以从gpt2-ml代码库提供的google drive地址:https://github.com/imcaspar/gpt2-ml#pretrained-model 下载Tensorflow版本,然后用本库代码把它转换成Pytorch版本。

henryxiao1997 commented 3 years ago

好的,尝试用过代码库中的脚本,不过在cuda 9.x的系统上会出现 core dump。我已经从百度网盘上下载到了。谢谢!~

ghosthamlet commented 3 years ago

不客气。cuda 9.x有可能不兼容,我只测试过cuda 10.2,但转换模型不需要用到GPU,所以可能是和其他环境配置不兼容。