INNER CODE UNIT · Python
__init__
shibing624/dialogbot · dialogbot/bot.py:14
def __init__(
self,
vocab_path=config.vocab_path,
search_model=config.search_model,
question_answer_path=config.question_answer_path,
context_response_path=config.context_response_path,
context=None,
model_name_or_path="shibing624/gpt2-dialogbot-base-chinese",
max_history_len=3,
max_len=128,
repetition_penalty=1.0,
temperature=1.0,
topk=8,
topp=0.0,
last_txt_len=100
):
self.context = context if context else []