geekyrakshit commited on
Commit
353a440
·
1 Parent(s): dc4625d

fix: catch byaldi import

Browse files
medrag_multi_modal/retrieval/colpali_retrieval.py CHANGED
@@ -2,7 +2,12 @@ import os
2
  from typing import Any, Optional
3
 
4
  import weave
5
- from byaldi import RAGMultiModalModel
 
 
 
 
 
6
  from PIL import Image
7
 
8
  import wandb
 
2
  from typing import Any, Optional
3
 
4
  import weave
5
+
6
+ try:
7
+ from byaldi import RAGMultiModalModel
8
+ except ImportError:
9
+ pass
10
+
11
  from PIL import Image
12
 
13
  import wandb