trial
This commit is contained in:
parent
b4cf5accdf
commit
ca125bb3cd
2
main.py
2
main.py
|
@ -139,7 +139,7 @@ if __name__ == "__main__":
|
||||||
nlu = NLU(query_rewriter, coref_resolver, intent_classifier, offensive_classifier, entity_extractor, ambig_classifier)
|
nlu = NLU(query_rewriter, coref_resolver, intent_classifier, offensive_classifier, entity_extractor, ambig_classifier)
|
||||||
|
|
||||||
#load retriever and generator
|
#load retriever and generator
|
||||||
retriever = SentenceTransformer('/models/BigRetriever/').to(device)
|
retriever = SentenceTransformer('/models/retriever/').to(device)
|
||||||
qa_generator = pipeline("text2text-generation", model="/models/train_qa", device=device_flag)
|
qa_generator = pipeline("text2text-generation", model="/models/train_qa", device=device_flag)
|
||||||
summ_generator = pipeline("text2text-generation", model="/models/train_summ", device=device_flag)
|
summ_generator = pipeline("text2text-generation", model="/models/train_summ", device=device_flag)
|
||||||
chat_generator = pipeline("text2text-generation", model="/models/train_chat", device=device_flag)
|
chat_generator = pipeline("text2text-generation", model="/models/train_chat", device=device_flag)
|
||||||
|
|
Loading…
Reference in New Issue