Ask for GPU pre-training checkpoint

[MathamPollard]

Could you provide GPU pre-training checkpoints? Run codes with CPU is too slow.

[dingkwang]

I also face this issue. The pre-trained model provided by Google Drive only supports CPU inference. I have to set os.environ["CUDA_VISIBLE_DEVICES"]="" in the inference.py to only use CPU.