Upload modeling_mplug_owl2.py with huggingface_hub

modeling_mplug_owl2.py

@@ -256,7 +256,7 @@ class MPLUGOwl2LlamaForCausalLM(LlamaForCausalLM, MPLUGOwl2MetaForCausalLM):
 
         self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
         self.preferential_ids_ = [id_[1] for id_ in self.tokenizer(["excellent","good","fair","poor","bad"])["input_ids"]]
-        self.weight_tensor = torch.Tensor([5.,4.,3.,2.,1.]).half().to(model.device)
+        self.weight_tensor = torch.Tensor([5.,4.,3.,2.,1.]).half().to(self.device)
 
         # Initialize weights and apply final processing
         self.post_init()