Spaces:

bokesyo
/

MiniCPM_Visual_Document_Retriever_Demo

Running on Zero

bokesyo commited on Jul 13

Commit

38c389d

•

1 Parent(s): 7264b3a

add cida

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,7 +121,7 @@ if __name__ == "__main__":
     from PIL import Image
     import torch
-    device = 'cpu'
     # Load model, be sure to substitute `model_path` by your model path
     model_path = 'RhapsodyAI/minicpm-visual-embedding-v0' # replace with your local model path

     from PIL import Image
     import torch
+    device = 'cuda:0'
     # Load model, be sure to substitute `model_path` by your model path
     model_path = 'RhapsodyAI/minicpm-visual-embedding-v0' # replace with your local model path