CogView2
CogView2 copied to clipboard
2.1.2+cu121的pytorch在load coglm的时候,时长过长
环境 : Driver Version: 535.129.03 CUDA Version: 12.2 sat: 0.4.9 (是最新版本的sat)
测试load coglm的速度的代码:
结果: torch version:1.13.1+cu117 load model需要的时间:209.14s
torch version: 2.1.2+cu121 load model需要的时间:527.55s
**另外有一个问题:是否有huggingface的coglm生成图片的script?因为我看到有huggingface的weights了https://huggingface.co/THUDM/CogView2 **