Spaces:

rynmurdock
/

generative_recsys

Running on L40S

rynmurdock commited on Jun 2, 2024

Commit

f1fda51

1 Parent(s): e4ec5a4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -582,7 +582,7 @@ def encode_space(x, im):
     print('encoded')
     print('pali_enc')
-    gemb = pali.vision_tower(im.to(dtype)).last_hidden_state
     print('pali_enced')
     return im_emb.detach().to(torch.float32), gemb.detach().to(torch.float32)

     print('encoded')
     print('pali_enc')
+    gemb = pali.vision_tower(im.to(dtype).to('cuda')).last_hidden_state
     print('pali_enced')
     return im_emb.detach().to(torch.float32), gemb.detach().to(torch.float32)