damerajee commited on
Commit
7b26342
·
verified ·
1 Parent(s): 493bcaf

Update vision_encoder.py

Browse files
Files changed (1) hide show
  1. vision_encoder.py +0 -1
vision_encoder.py CHANGED
@@ -18,5 +18,4 @@ class VisionEncoder(nn.Module):
18
  with torch.no_grad():
19
  pixel_values = self.vision_model(processed_images)
20
  image_features = pixel_values.last_hidden_state
21
- image_features = image_features
22
  return image_features
 
18
  with torch.no_grad():
19
  pixel_values = self.vision_model(processed_images)
20
  image_features = pixel_values.last_hidden_state
 
21
  return image_features