Epoching commited on
Commit
ead4891
1 Parent(s): c14d9ad

Update CrossEncoder/cross_encoder.py

Browse files
Files changed (1) hide show
  1. CrossEncoder/cross_encoder.py +2 -2
CrossEncoder/cross_encoder.py CHANGED
@@ -92,7 +92,7 @@ class CERank:
92
  def get_ranked_contexts(context_json, question):
93
 
94
  dirname = 'examples'
95
- model_path = '/data/actici/pretrained_weights/ms-marco-electra-base'
96
  max_length = 512
97
 
98
  # Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.
@@ -110,7 +110,7 @@ def get_ranked_contexts(context_json, question):
110
  def get_ranked_contexts_in_memory(contexts, question):
111
 
112
  dirname = 'examples'
113
- model_path = '/data/actici/pretrained_weights/ms-marco-electra-base'
114
  max_length = 512
115
 
116
  # Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.
 
92
  def get_ranked_contexts(context_json, question):
93
 
94
  dirname = 'examples'
95
+ model_path = 'ms-marco-electra-base'
96
  max_length = 512
97
 
98
  # Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.
 
110
  def get_ranked_contexts_in_memory(contexts, question):
111
 
112
  dirname = 'examples'
113
+ model_path = 'ms-marco-electra-base'
114
  max_length = 512
115
 
116
  # Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.