github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 20:39:59 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cb413ccf28a82602eedfeebb3be5224aa02486dc
ColossalAI/colossalai/autochunk/utils.py
Xuanlei Zhao 10c61de2f7 [autochunk] support vit (#3084) 
support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit
2023-03-10 10:23:26 +08:00

7.0 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink