Skip to content

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

@duchenzhuang

Description

@duchenzhuang

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions