mirror of
https://github.com/vladmandic/automatic
synced 2026-09-08 05:48:42 +02:00
Typing update
This commit is contained in:
@@ -459,7 +459,7 @@ def update_token_counter(text: str):
|
||||
|
||||
try:
|
||||
try:
|
||||
ids = getattr(tokenizer(prompt_list), 'input_ids', [])
|
||||
ids: list = getattr(tokenizer(prompt_list), 'input_ids', [])
|
||||
except TypeError:
|
||||
for p in prompt_list:
|
||||
ids.append(getattr(tokenizer(p), 'input_ids', []))
|
||||
|
||||
Reference in New Issue
Block a user