CogView2 icon indicating copy to clipboard operation
CogView2 copied to clipboard

2.1.2+cu121的pytorch在load coglm的时候,时长过长

Open duchenzhuang opened this issue 2 years ago • 0 comments

环境 : Driver Version: 535.129.03 CUDA Version: 12.2 sat: 0.4.9 (是最新版本的sat)

测试load coglm的速度的代码: image

结果: 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 **

duchenzhuang avatar Dec 25 '23 12:12 duchenzhuang