mirror of https://github.com/InternLM/InternLM
[Fix]: Update end token id in web_demo.py (#611)
parent
39ffce9090
commit
e408af7fb5
|
@ -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