Update src/LMdecoder.py

src/LMdecoder.py

@@ -81,11 +81,11 @@ class LMDecoder(nn.Module):
         
     def load_lm(self):
         ## ---------------------LM setting----------------------
-        self.tokenizer = AutoTokenizer.from_pretrained(self.decoder_type)
+        self.tokenizer = AutoTokenizer.from_pretrained('Tabgac/SpectPrompt/Llama-2-7b-hf')
         if self.tokenizer.pad_token is None:
             self.tokenizer.pad_token = self.tokenizer.eos_token
-        self.LMconfig = AutoConfig.from_pretrained(self.decoder_type, trust_remote_code=True )
-        self.lm = AutoModelForCausalLM.from_pretrained(self.decoder_type, trust_remote_code=True)
+        self.LMconfig = AutoConfig.from_pretrained('Tabgac/SpectPrompt/Llama-2-7b-hf', trust_remote_code=True )
+        self.lm = AutoModelForCausalLM.from_pretrained('Tabgac/SpectPrompt/Llama-2-7b-hf', trust_remote_code=True)
        
         
     def forward(self, input_ids, flatten_embs, attention_mask, labels, **kwargs):