Update modeling_mplug_owl2.py
Browse files- modeling_mplug_owl2.py +2 -2
modeling_mplug_owl2.py
CHANGED
@@ -308,8 +308,8 @@ class MPLUGOwl2LlamaForCausalLM(LlamaForCausalLM, MPLUGOwl2MetaForCausalLM):
|
|
308 |
def __init__(self, config):
|
309 |
super(LlamaForCausalLM, self).__init__(config)
|
310 |
self.model = MPLUGOwl2LlamaModel(config)
|
311 |
-
self.tokenizer = AutoTokenizer.from_pretrained("
|
312 |
-
self.image_processor = CLIPImageProcessor.from_pretrained("
|
313 |
|
314 |
self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
|
315 |
self.preferential_ids_ = [id_[1] for id_ in self.tokenizer(["inferior", "worse", "similar", "better", "superior"])["input_ids"]]
|
|
|
308 |
def __init__(self, config):
|
309 |
super(LlamaForCausalLM, self).__init__(config)
|
310 |
self.model = MPLUGOwl2LlamaModel(config)
|
311 |
+
self.tokenizer = AutoTokenizer.from_pretrained("q-future/Compare2Score")
|
312 |
+
self.image_processor = CLIPImageProcessor.from_pretrained("q-future/Compare2Score")
|
313 |
|
314 |
self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
|
315 |
self.preferential_ids_ = [id_[1] for id_ in self.tokenizer(["inferior", "worse", "similar", "better", "superior"])["input_ids"]]
|