Skip to content

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

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