DataCanvas
/

MMAlaya

text-generation

Model card Files Files and versions

bingwork commited on Jan 23, 2024

Commit

60f50de

·

verified ·

1 Parent(s): a2561ec

Upload mmalaya_arch.py

Files changed (1) hide show

mmalaya_arch.py +2 -2

mmalaya_arch.py CHANGED Viewed

@@ -306,8 +306,8 @@ class MMAlayaMetaForCausalLM(ABC):
             torch.tensor(input_ids, dtype=torch.long).unsqueeze(0),
             )
         # 加载图像
-        image_processor = model.get_vision_tower().image_processor
-        image_tensor = image_processor(image, return_tensors='pt')['pixel_values'].half()
         if return_tensors is not None:
             if return_tensors == 'pt':

             torch.tensor(input_ids, dtype=torch.long).unsqueeze(0),
             )
         # 加载图像
+        image_tensor = self.get_vision_tower().image_processor(
+            image, return_tensors='pt')['pixel_values'].half()
         if return_tensors is not None:
             if return_tensors == 'pt':