ColossalAI/colossalai/gemini/update/__init__.py

4 lines
141 B
Python

from .chunkv2 import ChunkV2
from .chunk_mgrv2 import ChunkManagerV2
from .search_utils import clasify_params, search_chunk_configuration