SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 36 Python

[autochunk] support vit (#3084)

support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit
X
Xuanlei Zhao committed
10c61de2f7ba2f58de70d24de862b6c9bb3cd3f3
Parent: e58a3c8
Committed by GitHub <noreply@github.com> on 3/10/2023, 2:23:26 AM