mirror of https://github.com/InternLM/InternLM
Update end token id inweb_demo.py
parent
112ed18575
commit
1b6a66ad9c
|
@ -103,7 +103,7 @@ def main():
|
||||||
model=model,
|
model=model,
|
||||||
tokenizer=tokenizer,
|
tokenizer=tokenizer,
|
||||||
prompt=real_prompt,
|
prompt=real_prompt,
|
||||||
additional_eos_token_id=103028,
|
additional_eos_token_id=92542,
|
||||||
**asdict(generation_config),
|
**asdict(generation_config),
|
||||||
):
|
):
|
||||||
# Display robot response in chat message container
|
# Display robot response in chat message container
|
||||||
|
|
Loading…
Reference in New Issue