Spaces:
Sleeping
Sleeping
Commit
·
c063304
1
Parent(s):
9595675
upfate: CalPaliRetriever methods to align with other retrievers
Browse files
medrag_multi_modal/retrieval/colpali_retrieval.py
CHANGED
@@ -59,7 +59,7 @@ class CalPaliRetriever(weave.Model):
|
|
59 |
from medrag_multi_modal.retrieval import CalPaliRetriever
|
60 |
|
61 |
weave.init(project_name="ml-colabs/medrag-multi-modal")
|
62 |
-
retriever = CalPaliRetriever.
|
63 |
index_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy:v0",
|
64 |
metadata_dataset_name="grays-anatomy-images:v0",
|
65 |
data_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy-images:v1",
|
@@ -100,7 +100,7 @@ class CalPaliRetriever(weave.Model):
|
|
100 |
)
|
101 |
|
102 |
@classmethod
|
103 |
-
def
|
104 |
cls,
|
105 |
index_artifact_name: str,
|
106 |
metadata_dataset_name: str,
|
|
|
59 |
from medrag_multi_modal.retrieval import CalPaliRetriever
|
60 |
|
61 |
weave.init(project_name="ml-colabs/medrag-multi-modal")
|
62 |
+
retriever = CalPaliRetriever.from_wandb_artifact(
|
63 |
index_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy:v0",
|
64 |
metadata_dataset_name="grays-anatomy-images:v0",
|
65 |
data_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy-images:v1",
|
|
|
100 |
)
|
101 |
|
102 |
@classmethod
|
103 |
+
def from_wandb_artifact(
|
104 |
cls,
|
105 |
index_artifact_name: str,
|
106 |
metadata_dataset_name: str,
|