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

[shardformer] test all optimizations (#4399)

Browse Source 
[shardformer] test all optimizations

[shardformer] test all optimizations

[shardformer] test all optimizations
This commit is contained in:
flybird1111
2023-08-10 13:59:30 +08:00
committed by Hongxin Liu
parent 7a3dfd0c64
commit d2cd48e0be
4 changed files with 59 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
requirements/requirements-test.txt
View File

@@ -19,4 +19,4 @@ ninja
flash_attn>=2.0
datasets
ninja
flash-attn
flash-attn>=2.0