[refactor] move chunk and chunkmgr to directory gemini (#1182)

colossalai/zero/utils/zero_hook_v2.py

@@ -1,6 +1,6 @@
 import torch
 from colossalai.tensor.param_op_hook import ParamOpHook
-from colossalai.tensor.chunk import ChunkManager, TensorState
+from colossalai.gemini import TensorState
 from enum import Enum
 from typing import List
 from contextlib import contextmanager