fix typo

colossalai/autochunk/search_chunk.py

@@ -1,6 +1,6 @@
 import copy
 
-from .estiamte_memory import EstimateMemory
+from .estimate_memory import EstimateMemory
 from .reorder_graph import ReorderGraph
 from .select_chunk import SelectChunk
 from .trace_flow import TraceFlow

colossalai/autochunk/select_chunk.py

@@ -1,4 +1,4 @@
-from .estiamte_memory import EstimateMemory
+from .estimate_memory import EstimateMemory
 from .reorder_graph import ReorderGraph
 from .trace_index import TraceIndex
 from .utils import is_non_compute_node