Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -585,7 +585,7 @@ def encode_space(x, im):
|
|
| 585 |
gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
|
| 586 |
|
| 587 |
print('pali_enced')
|
| 588 |
-
return im_emb.
|
| 589 |
|
| 590 |
demo.launch(share=True,)
|
| 591 |
|
|
|
|
| 585 |
gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
|
| 586 |
|
| 587 |
print('pali_enced')
|
| 588 |
+
return im_emb.to(torch.float32), gemb.to(torch.float32)
|
| 589 |
|
| 590 |
demo.launch(share=True,)
|
| 591 |
|