环境 :
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 **
环境 :
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 **