Update end token id inweb_demo.py

pull/611/head
LYMDLUT 2024-01-18 04:04:52 +08:00 committed by GitHub
parent 112ed18575
commit 1b6a66ad9c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -103,7 +103,7 @@ def main():
model=model,
tokenizer=tokenizer,
prompt=real_prompt,
additional_eos_token_id=103028,
additional_eos_token_id=92542,
**asdict(generation_config),
):
# Display robot response in chat message container