qnguyen3 commited on
Commit
daedff0
1 Parent(s): f93b9e6

Update modeling_llava_qwen2.py

Browse files
Files changed (1) hide show
  1. modeling_llava_qwen2.py +1 -1
modeling_llava_qwen2.py CHANGED
@@ -662,7 +662,7 @@ class LlavaMetaForCausalLM(ABC):
662
  return self.get_model().get_vision_tower()
663
 
664
  def encode_images(self, images):
665
- image_features = self.get_model().get_vision_tower().to(decice="cuda:0", dtype=torch.float)(images)
666
  image_features = self.get_model().mm_projector.to(device="cuda:0", dtype=torch.float)(image_features)
667
  return image_features
668
 
 
662
  return self.get_model().get_vision_tower()
663
 
664
  def encode_images(self, images):
665
+ image_features = self.get_model().get_vision_tower().to(device="cuda:0")(images)
666
  image_features = self.get_model().mm_projector.to(device="cuda:0", dtype=torch.float)(image_features)
667
  return image_features
668