rynmurdock commited on
Commit
82bf92e
·
1 Parent(s): f1fda51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.detach().to(torch.float32), gemb.detach().to(torch.float32)
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