github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-09 13:00:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

[shardformer] support whisper (#4212)

Browse Source 
* support whisper

* fix bug in vocabembedding

* support downstream model of whisper

* update readme
This commit is contained in:
FoolPlayer
2023-07-17 14:25:32 +08:00
committed by Hongxin Liu
parent dd2bf02679
commit 9ee4ebea83
7 changed files with 443 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/kit/model_zoo/transformers/__init__.py
View File

@@ -7,3 +7,4 @@ from .opt import *
from .sam import *
from .t5 import *
from .vit import *
from .whisper import *