resolve comments

pytorch · Oct 15, 2024 · 402c9b0 · 402c9b0
1 parent c57ab06
commit 402c9b0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py b/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py
@@ -640,7 +640,8 @@ def run(
 
         self._construct_trt_network_def()
 
-        self._save_weight_mapping()
+        if not self.compilation_settings.immutable_weights:
+            self._save_weight_mapping()
 
         build_engine_start_time = datetime.now()
         _LOGGER.info("Not found cached TRT engines. Start building engine.")
@@ -659,7 +660,7 @@ def run(
         assert serialized_engine
 
         _LOGGER.info(
-            f"Build weight-stripped TRT engine elapsed time: {datetime.now() - build_engine_start_time}"
+            f"Build TRT engine elapsed time: {datetime.now() - build_engine_start_time}"
         )
         _LOGGER.info(f"TRT Engine uses: {serialized_engine.nbytes} bytes of Memory")