Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Better warpers logging

Browse Source
This commit is contained in:
oobabooga 2024-02-06 07:09:21 -08:00
parent 494cc3c5b0
commit 3add2376cd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/sampler_hijack.py
View File

@ -431,6 +431,7 @@ def get_logits_warper_patch(self, generation_config):
if shared.args.verbose: if shared.args.verbose:
logger.info("WARPERS=") logger.info("WARPERS=")
pprint.PrettyPrinter(indent=4, sort_dicts=False).pprint([x.__class__.__name__ for x in warpers]) pprint.PrettyPrinter(indent=4, sort_dicts=False).pprint([x.__class__.__name__ for x in warpers])
print()
if normalize is not None: if normalize is not None:
warpers.append(normalize) warpers.append(normalize)