teowu commited on
Commit
ffa96e2
1 Parent(s): f702052

Upload modeling_mplug_owl2.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_mplug_owl2.py +1 -1
modeling_mplug_owl2.py CHANGED
@@ -272,7 +272,7 @@ class MPLUGOwl2LlamaForCausalLM(LlamaForCausalLM, MPLUGOwl2MetaForCausalLM):
272
  images = [expand2square(img, tuple(int(x*255) for x in self.image_processor.image_mean)) for img in images]
273
  image_tensor = self.image_processor.preprocess(images, return_tensors="pt")["pixel_values"].half().to(self.device)
274
 
275
- return self.model.generate(input_ids, images=image_tensor, streamer=self.streamer, **generate_kwargs)
276
  def score(self, images,
277
  task_: str = "quality",
278
  input_: str = "image",
 
272
  images = [expand2square(img, tuple(int(x*255) for x in self.image_processor.image_mean)) for img in images]
273
  image_tensor = self.image_processor.preprocess(images, return_tensors="pt")["pixel_values"].half().to(self.device)
274
 
275
+ return self.generate(input_ids, images=image_tensor, streamer=self.streamer, **generate_kwargs)
276
  def score(self, images,
277
  task_: str = "quality",
278
  input_: str = "image",