From c1752b3d41e1bdfeaa66256e7640f18aab92cb83 Mon Sep 17 00:00:00 2001 From: Jimmy Zhang Date: Sat, 20 Apr 2024 15:31:39 -0700 Subject: [PATCH] cleanup code Signed-off-by: Jimmy Zhang --- nemo/export/trt_llm/nemo/convert.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/nemo/export/trt_llm/nemo/convert.py b/nemo/export/trt_llm/nemo/convert.py index 5ffe7e9e0894..c87d2c7f0e21 100644 --- a/nemo/export/trt_llm/nemo/convert.py +++ b/nemo/export/trt_llm/nemo/convert.py @@ -520,9 +520,5 @@ def save_tranpose(val, key, shared=False): elif "vocab_embedding" in key or "lm_head.weight" in key: save(key, gpu_val) else: - if torch.cuda.current_device() == 0: - import pdb - pdb.set_trace() - torch.distributed.barrier() raise RuntimeError(f"{key} not handled by NeMo->TRTLLM converter!")