line

mosaicml · Apr 10, 2024 · 379969b · 379969b
1 parent 0ee1fad
commit 379969b
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/composer/trainer/mosaic_fsdp_utils.py b/composer/trainer/mosaic_fsdp_utils.py
@@ -1137,10 +1137,6 @@ def _shard_orig_param_state(
         fqn: str,
         optim_state: Dict[str, Any],
     ) -> Dict[str, Any]:
-        """
-        Shard the optimizer state for the original parameter with the name ``fqn``.
-        This API should only be used when ``use_orig_params`` is True.
-        """
         if not optim_state:
             return {}
         fsdp_state = fsdp_param_info.state