Skip to content

Commit 6691c91

Browse files
committed
Fix for some tokenizer_config formats
1 parent a203a0b commit 6691c91

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

exllamav2/tokenizer/tokenizer.py

+10-1
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,16 @@ def get_default_token_id(config_key: str, current: int | None, default: int):
168168
if self.tokenizer_config_dict is not None and config_key in self.tokenizer_config_dict:
169169
st = self.tokenizer_config_dict[config_key]
170170
if st is None: return None
171-
return self.tokenizer_model.piece_to_id(st)
171+
if isinstance(st, dict):
172+
stc: str | None = st.get("content", None)
173+
if stc is None:
174+
return None
175+
else:
176+
return self.tokenizer_model.piece_to_id(stc)
177+
elif isinstance(st, str):
178+
return self.tokenizer_model.piece_to_id(st)
179+
else:
180+
return None
172181
else:
173182
return default
174183

0 commit comments

Comments
 (0)