H4nwei commited on
Commit
917f39d
1 Parent(s): d56984d

Update modeling_mplug_owl2.py

Browse files
Files changed (1) hide show
  1. 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("VQA-CityU/Compare2Score_1")
312
- self.image_processor = CLIPImageProcessor.from_pretrained("VQA-CityU/Compare2Score_1")
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"]]