Adding f-string to validation error which is missing (#6748)
This commit is contained in:
parent
f4f8a9d892
commit
545146349c
@ -107,7 +107,7 @@ def _image_token_str(model_config: ModelConfig,
|
|||||||
return tokenizer.decode(model_config.hf_config.image_token_index)
|
return tokenizer.decode(model_config.hf_config.image_token_index)
|
||||||
if model_type == "chameleon":
|
if model_type == "chameleon":
|
||||||
return "<image>"
|
return "<image>"
|
||||||
raise TypeError("Unknown model type: {model_type}")
|
raise TypeError(f"Unknown model type: {model_type}")
|
||||||
|
|
||||||
|
|
||||||
# TODO: Let user specify how to insert image tokens into prompt
|
# TODO: Let user specify how to insert image tokens into prompt
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user