menikev commited on
Commit
369c8e0
·
verified ·
1 Parent(s): 0a8b77c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ pipe = pipeline(
18
  model="microsoft/DialoGPT-medium",
19
  device_map="auto",
20
  max_new_tokens=150,
21
- temperature=0.1,
22
- do_sample=False
23
  )
24
 
25
  llm = HuggingFacePipeline(pipeline=pipe)
 
18
  model="microsoft/DialoGPT-medium",
19
  device_map="auto",
20
  max_new_tokens=150,
21
+ do_sample=False,
22
+ pad_token_id=tokenizer.eos_token_id
23
  )
24
 
25
  llm = HuggingFacePipeline(pipeline=pipe)