search

wilson1yan / VideoGPT

MIT License
962 stars 115 forks source link

deepspeed version #33

Closed yrcong closed 2 years ago

yrcong commented 2 years ago

Dear authors,

which version of deepspeed have you used? I am using deepspeed = 0.5.10 and get error with from deepspeed.ops.sparse_attention import MatMul, Softmax

I import the module with from deepspeed.ops.sparse_attention.matmul import MatMul. However, IndexError happens in the forward.

best

wilson1yan commented 2 years ago

Hi,

Perhaps #14 might be of help?

yrcong commented 2 years ago

Thx for the reply. I upgrade Triton version and it works:)