turn off print mem

2022-12-27 15:01:58 +08:00 · 2022-12-27 15:01:58 +08:00 · cb2dd1a106
parent a2b4755ce9
commit cb2dd1a106
1 changed files with 4 additions and 1 deletions
--- a/chunk_codegen.py
+++ b/chunk_codegen.py
@ -1750,10 +1750,13 @@ class ChunkRegionSearch(object):
                _,
                active_node,
            ) = self.memory_estimator.estimate_chunk_inference_mem(
-                self.index_tracer.node_list, chunk_infos, print_mem=True
+                self.index_tracer.node_list, chunk_infos
            )
            if self._stop_search(init_mem_peak, mem_peak):
                break
+        # self.memory_estimator.estimate_chunk_inference_mem(
+        #     self.index_tracer.node_list, chunk_infos, print_mem=True
+        # )
        return chunk_infos