rk68 commited on
Commit
eedb8bc
1 Parent(s): f8139ba

Update retrievers.py

Browse files
Files changed (1) hide show
  1. 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 = 'mps', # Set to 'mps' as the default device
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