From 2c2084be535e1888e933ee666fae80bdb66b5415 Mon Sep 17 00:00:00 2001 From: Daniel Huang Date: Wed, 11 Dec 2024 15:12:22 -0800 Subject: [PATCH] Added arctic model to supported models for autotp Signed-off-by: Daniel Huang --- docs/_tutorials/automatic-tensor-parallelism.md | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/_tutorials/automatic-tensor-parallelism.md b/docs/_tutorials/automatic-tensor-parallelism.md index d5a08b27bf4d..6488f9b718fe 100755 --- a/docs/_tutorials/automatic-tensor-parallelism.md +++ b/docs/_tutorials/automatic-tensor-parallelism.md @@ -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