Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4789 from awsr/fix-counter

Browse Source 
Fix token estimate counter
This commit is contained in:
Vladimir Mandic
2026-04-23 13:25:23 +02:00
committed by GitHub
parent 40e550f1c3 1d5154e125
commit f3bfa4d92c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ui_common.py
+2 -1
View File
@@ -507,9 +507,10 @@ def update_token_counter(text: str):
return gr.update(value=f"<span class='gr-box gr-text-input'>??/{max_length}</span>", visible=True)
token_counts = [len(group) - int(has_bos_token) - int(has_eos_token) for group in ids]
token_counts = [tc for tc in token_counts if tc > 0]
if len(token_counts) > 1:
visible = True
count_formatted = f"{token_counts}/{sum(token_counts)}"
count_formatted = f"{token_counts} {sum(token_counts)}"
elif len(token_counts) == 1 and token_counts[0] > 0:
visible = True
count_formatted = str(token_counts[0])