Spaces:
Running
on
A10G
Running
on
A10G
Commit
·
82bf92e
1
Parent(s):
f1fda51
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 |
|