diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp
index 4336327d3..1adb835b4 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py
index 1c6a0d477..3a74b8bba 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py
@@ -77,6 +77,8 @@ from icefall.utils import (
     write_error_stats,
 )
 
+from train_lora import LoRAHook
+
 LOG_EPS = math.log(1e-10)
 
 
@@ -681,8 +683,10 @@ def main():
     logging.info("About to create model")
     model = get_transducer_model(params)
 
-    if params.model_name:
+    if '.pt' in params.model_name:
         load_checkpoint(f"{params.exp_dir}/{params.model_name}", model)
+    elif 'lora' in params.model_name:
+        
     else:
         if not params.use_averaged_model:
             if params.iter > 0: