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

Don't print torch tensors with --verbose

Browse Source
This commit is contained in:
oobabooga 2024-01-07 10:35:55 -08:00
parent cf820c69c5
commit b8a0b3f925
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/text_generation.py
View File

@ -346,7 +346,8 @@ def generate_reply_HF(question, original_question, seed, state, stopping_strings
if shared.args.verbose: if shared.args.verbose:
logger.info("GENERATE_PARAMS=") logger.info("GENERATE_PARAMS=")
pprint.PrettyPrinter(indent=4, sort_dicts=False).pprint(generate_params) filtered_params = {key: value for key, value in generate_params.items() if not isinstance(value, torch.Tensor)}
pprint.PrettyPrinter(indent=4, sort_dicts=False).pprint(filtered_params)
print() print()
t0 = time.time() t0 = time.time()