Update retrievers.py
Browse files- retrievers.py +1 -1
retrievers.py
CHANGED
@@ -23,7 +23,7 @@ class PARetriever(BaseRetriever):
|
|
23 |
mode: str = "OR",
|
24 |
rewriter: bool = True,
|
25 |
classifier_model: Optional[str] = None, # Optional classifier model
|
26 |
-
device: str = '
|
27 |
reranker_model_name: Optional[str] = None, # Model name for SentenceTransformerRerank
|
28 |
verbose: bool = False, # Verbose flag
|
29 |
fixed_params: Optional[dict] = None, # New parameter to pass in fixed parameters
|
|
|
23 |
mode: str = "OR",
|
24 |
rewriter: bool = True,
|
25 |
classifier_model: Optional[str] = None, # Optional classifier model
|
26 |
+
device: str = 'cpu', # Device to CPU for huggingface demo
|
27 |
reranker_model_name: Optional[str] = None, # Model name for SentenceTransformerRerank
|
28 |
verbose: bool = False, # Verbose flag
|
29 |
fixed_params: Optional[dict] = None, # New parameter to pass in fixed parameters
|