KvrParaskevi commited on
Commit
9af9202
1 Parent(s): cfc8480

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +1 -1
chatbot.py CHANGED
@@ -48,7 +48,7 @@ def load_pipeline():
48
  pipe = pipeline("text-generation",
49
  model= model,
50
  tokenizer = tokenizer,
51
- max_tokens = 252,top_k = 30, early_stopping=True,
52
  temperature = 0.1,repetition_penalty = 1.03)
53
 
54
  llm = HuggingFacePipeline(pipeline = pipe)
 
48
  pipe = pipeline("text-generation",
49
  model= model,
50
  tokenizer = tokenizer,
51
+ max_new_tokens = 252,top_k = 30, early_stopping=True,
52
  temperature = 0.1,repetition_penalty = 1.03)
53
 
54
  llm = HuggingFacePipeline(pipeline = pipe)