Upload modeling_mplug_owl2.py with huggingface_hub
Browse files- 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.
|
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",
|