remove unpack ckpt nemo

Signed-off-by: Jimmy Zhang <jiemingz@nvidia.com>
JimmyZhang12 · Apr 24, 2024 · 10274c9 · 10274c9
1 parent b120833
commit 10274c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/export/tensorrt_llm.py b/nemo/export/tensorrt_llm.py
@@ -33,7 +33,7 @@
 from nemo.export.trt_llm.nemo_utils import get_tokenzier, nemo_llm_model_to_model_config, nemo_llm_to_model_config
 from nemo.export.trt_llm.tensorrt_llm_build import build_and_save_engine
 from nemo.export.trt_llm.tensorrt_llm_run import generate, generate_streaming, load, load_refit, create_gpt_session
-from nemo.export.trt_llm.utils import is_nemo_file, unpack_nemo_ckpt
+from nemo.export.trt_llm.utils import is_nemo_file
 
 use_deploy = True
 try: