Camil Ziane
commited on
Commit
•
340a0f3
1
Parent(s):
eee8990
fix: slow_conv2d_cpu not implemented for Half
Browse files
TinyLLaVA_Factory/tinyllava/model/vision_tower/base.py
CHANGED
|
@@ -45,6 +45,7 @@ class VisionTower(nn.Module):
|
|
| 45 |
|
| 46 |
|
| 47 |
def forward(self, x, **kwargs):
|
|
|
|
| 48 |
image_features = self._vision_tower(x, output_hidden_states=True)
|
| 49 |
image_features = image_features.hidden_states[kwargs.get('vision_feature_layer', -2)]
|
| 50 |
|
|
|
|
| 45 |
|
| 46 |
|
| 47 |
def forward(self, x, **kwargs):
|
| 48 |
+
x = x.to(torch.float32)
|
| 49 |
image_features = self._vision_tower(x, output_hidden_states=True)
|
| 50 |
image_features = image_features.hidden_states[kwargs.get('vision_feature_layer', -2)]
|
| 51 |
|