kiranr commited on
Commit
3dccb13
·
verified ·
1 Parent(s): dedd647

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -71,7 +71,7 @@ input_ids = tokenizer.apply_chat_template(
71
 
72
  gen_conf = {
73
  "max_new_tokens": 256,
74
- "eos_token_id": [tokenizer.eos_token_id, tokenizer.convert_tokens_to_ids("<|eot_id|>")],
75
  "temperature": 0.7,
76
  "top_p": 0.9,
77
  }
 
71
 
72
  gen_conf = {
73
  "max_new_tokens": 256,
74
+ "eos_token_id": tokenizer.eos_token_id,
75
  "temperature": 0.7,
76
  "top_p": 0.9,
77
  }