Skip to content

GitLab

Commit d2d5086e authored by Mostofa Patwary's avatar Mostofa Patwary
Browse files

DPR ongoing

parent 220637f9
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
megatron/indexer.py
View file @ d2d5086e
...@@ -49,15 +49,15 @@ class IndexBuilder(object): ...@@ -49,15 +49,15 @@ class IndexBuilder(object):
if self.biencoder_shared_query_context_model: if self.biencoder_shared_query_context_model:
only_context_model = False only_context_model = False
#model = get_model(lambda: biencoder_model_provider(only_context_model \ model = get_model(lambda: biencoder_model_provider(only_context_model \
# = only_context_model, biencoder_shared_query_context_model = \
# self.biencoder_shared_query_context_model, \
# pre_process=self.pre_process, post_process=self.post_process))
model = biencoder_model_provider(only_context_model \
= only_context_model, biencoder_shared_query_context_model = \ = only_context_model, biencoder_shared_query_context_model = \
self.biencoder_shared_query_context_model, \ self.biencoder_shared_query_context_model, \
pre_process=self.pre_process, post_process=self.post_process) pre_process=self.pre_process, post_process=self.post_process))
#model = biencoder_model_provider(only_context_model \
# = only_context_model, biencoder_shared_query_context_model = \
# self.biencoder_shared_query_context_model, \
# pre_process=self.pre_process, post_process=self.post_process)
self.model = load_biencoder_checkpoint(model, self.model = load_biencoder_checkpoint(model,
only_context_model=only_context_model) only_context_model=only_context_model)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment