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):
if self.biencoder_shared_query_context_model:
only_context_model = False
#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 \
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)
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,
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