Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-30 14:10:14 +01:00
Code Issues Packages Projects Releases Wiki Activity

Verbose should always print special tokens in input (#1707)

Browse Source
This commit is contained in:
Carl Kenner 2023-05-02 13:54:56 +09:30 committed by GitHub
parent 0df0b2d0f9
commit 2f1a2846d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/text_generation.py
View File

@ -229,7 +229,7 @@ def generate_reply(question, state, eos_token=None, stopping_strings=[]):
output = input_ids[0] output = input_ids[0]
cuda = not any((shared.args.cpu, shared.args.deepspeed, shared.args.flexgen)) cuda = not any((shared.args.cpu, shared.args.deepspeed, shared.args.flexgen))
if shared.args.verbose: if shared.args.verbose:
print(f'\n\n{decode(input_ids[0], state["skip_special_tokens"])}\n--------------------\n') print(f'\n\n{decode(input_ids[0], False)}\n--------------------\n')
# Find the eos tokens # Find the eos tokens
eos_token_ids = [shared.tokenizer.eos_token_id] if shared.tokenizer.eos_token_id is not None else [] eos_token_ids = [shared.tokenizer.eos_token_id] if shared.tokenizer.eos_token_id is not None else []