rynmurdock commited on
Commit
c9470c3
·
verified ·
1 Parent(s): 9b8f9c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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