Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -582,7 +582,7 @@ def encode_space(x, im):
|
|
| 582 |
print('encoded')
|
| 583 |
|
| 584 |
print('pali_enc')
|
| 585 |
-
gemb = pali.vision_tower(im.to(dtype)).last_hidden_state
|
| 586 |
|
| 587 |
print('pali_enced')
|
| 588 |
return im_emb.detach().to(torch.float32), gemb.detach().to(torch.float32)
|
|
|
|
| 582 |
print('encoded')
|
| 583 |
|
| 584 |
print('pali_enc')
|
| 585 |
+
gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
|
| 586 |
|
| 587 |
print('pali_enced')
|
| 588 |
return im_emb.detach().to(torch.float32), gemb.detach().to(torch.float32)
|