update doc

615e7e68 · oahzxl · 7d4abaa5 · 615e7e68
Commit 615e7e68 authored Jan 10, 2023 by oahzxl
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/autochunk/search_chunk.py colossalai/autochunk/search_chunk.py +1 -1

No files found.
--- a/colossalai/autochunk/search_chunk.py
+++ b/colossalai/autochunk/search_chunk.py
@@ -88,7 +88,7 @@ class SearchChunk(object):
        Args:
            active_node (List): active node status for every node
            peak_node (Node): peak memory node
-            chunk_regions (List): chunk region info
+            chunk_regions (List): chunk region infos

        Returns:
            chunk_region_start (int)