Spaces:

p3nguknight
/

colpali-pixtral

Runtime error

p3nguknight commited on Jul 12

Commit

efb1452

1 Parent(s): 58bfa53

Try with use_fast=True

Files changed (1) hide show

app.py CHANGED Viewed

@@ -98,7 +98,7 @@ def retrieve(query: str, ds, images, k):
     model.load_adapter(COLPALI_MODEL_PATH)
     model = model.eval()
-    processor = ColPaliProcessor.from_pretrained(COLPALI_MODEL_PATH)
     qs = []
     with torch.no_grad():

     model.load_adapter(COLPALI_MODEL_PATH)
     model = model.eval()
+    processor = ColPaliProcessor.from_pretrained(COLPALI_MODEL_PATH, use_fast=True)
     qs = []
     with torch.no_grad():