[chore] refactor & sync

2025-09-05 11:02:05 +00:00 · 2024-05-16 07:22:10 +00:00
parent 4148ceed9f
commit 2e68eebdfe
7 changed files with 82 additions and 46 deletions
--- a/colossalai/zero/gemini/chunk/chunk.py
+++ b/colossalai/zero/gemini/chunk/chunk.py
@@ -567,6 +567,7 @@ class Chunk:
        return self is __o

    def __repr__(self, detailed: bool = True):
+        return f"Chunk({self.count_id})"
        output = [
            "Chunk Information:\n",
            "\tchunk size: {}, chunk dtype: {}, process group size: {}\n".format(