github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_pipeline/test_pipeline_utils
History
Jianghai 8739aa7fa0
[shardformer] Pipeline/whisper (#4456) 
* add some base tests and policies

* finish whisper base model

* add conditional generation

* finish basic tests

* whisper

* finish whisper

* finish whisper

* del useless  whisper test

* fix

* add argmin to replace

* finish revision
 		2023-08-18 21:29:25 +08:00
..
test_t5_pipeline_utils.py [shardformer] Pipeline/whisper (#4456) 2023-08-18 21:29:25 +08:00
test_whisper_pipeline_utils.py [shardformer] Pipeline/whisper (#4456) 2023-08-18 21:29:25 +08:00