Skip to content

Commit

Permalink
Added arctic model to supported models for autotp
Browse files Browse the repository at this point in the history
Signed-off-by: Daniel Huang <[email protected]>
  • Loading branch information
pi314ever committed Dec 11, 2024
1 parent 5164fb4 commit 2c2084b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions docs/_tutorials/automatic-tensor-parallelism.md
Original file line number Diff line number Diff line change
Expand Up @@ -121,6 +121,7 @@ The following results were collected using V100 SXM2 32GB GPUs.
The following model families have been successfully tested with automatic tensor parallelism. Other models may work but have not been tested yet.

- albert
- arctic
- baichuan
- bert
- bigbird_pegasus
Expand Down

0 comments on commit 2c2084b

Please sign in to comment.