AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Remove debug statements

Browse Source
This commit is contained in:
oobabooga 2024-09-27 15:02:36 -07:00
parent 3f5d8a05b5
commit 203783a58c
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sampler_hijack.py
View File

@ -546,10 +546,8 @@ def get_logits_processor_patch(self, **kwargs):
# Return a large value if class name is not mapped or if the mapped nickname is not in priority
if class_name not in class_name_to_nickname or class_name_to_nickname[class_name] not in sampler_priority:
print("======>", class_name, "inf")
return float('inf')
print("======>", class_name, sampler_priority.index(class_name_to_nickname[class_name]))
# Return the index of the nickname in the priority list for sorting
return sampler_priority.index(class_name_to_nickname[class_name])