cleanup code

Signed-off-by: Jimmy Zhang <jiemingz@nvidia.com>
JimmyZhang12 · Apr 20, 2024 · c1752b3 · c1752b3
1 parent d6e76f9
commit c1752b3
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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!")