sino commited on
Commit
5fb0bcb
1 Parent(s): ff4fdee

Update LMdecoder.py

Browse files
Files changed (1) hide show
  1. LMdecoder.py +2 -2
LMdecoder.py CHANGED
@@ -84,8 +84,8 @@ class LMDecoder(nn.Module):
84
  self.tokenizer = AutoTokenizer.from_pretrained(self.decoder_type)
85
  if self.tokenizer.pad_token is None:
86
  self.tokenizer.pad_token = self.tokenizer.eos_token
87
- self.LMconfig = AutoConfig.from_pretrained(self.decoder_type, trust_remote_code=True )
88
- self.lm = AutoModelForCausalLM.from_pretrained(self.decoder_type, trust_remote_code=True)
89
 
90
 
91
  def forward(self, input_ids, flatten_embs, attention_mask, labels, **kwargs):
 
84
  self.tokenizer = AutoTokenizer.from_pretrained(self.decoder_type)
85
  if self.tokenizer.pad_token is None:
86
  self.tokenizer.pad_token = self.tokenizer.eos_token
87
+ self.LMconfig = AutoConfig.from_pretrained(self.decoder_type, token='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')
88
+ self.lm = AutoModelForCausalLM.from_pretrained(self.decoder_type, token='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')
89
 
90
 
91
  def forward(self, input_ids, flatten_embs, attention_mask, labels, **kwargs):