Update src/LMdecoder.py

src/LMdecoder.py

@@ -81,6 +81,7 @@ class LMDecoder(nn.Module):
         
     def load_lm(self):
         ## ---------------------LM setting----------------------
+        __import__('pdb').set_trace()
         self.tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-hf', token='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')
         if self.tokenizer.pad_token is None:
             self.tokenizer.pad_token = self.tokenizer.eos_token