Time Cost for image embedding

[Pride-Huang]

Hi， Thanks for your great work. When I embed a image with size 1376*768 on a H100 GPU, I cost 40 seconds. Is this a normal time cost?

[dingkun-ldk]

Not normal. Please check your PyTorch version. You can refer to this issue:

pytorch/pytorch#166122

[Pride-Huang]

Not normal. Please check your PyTorch version. You can refer to this issue:

pytorch/pytorch#166122

Thanks, I make torch 2.9.0 to 2.8.0, the time cost from 40.0s to 1.1s. Is this normal?