better description

mosaicml · Sep 26, 2024 · 7ac37bc · 7ac37bc
1 parent d5779c7
commit 7ac37bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py
@@ -701,7 +701,7 @@ def _validate_cfg(icl_cfg: dict[str, Any]):
     return evaluators, logger_keys
 
 
-@experimental_function('tp_strategies')
+@experimental_function('Tensor Parallelism')
 def build_tp_strategies(
     name: str,
     model: ComposerModel,