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