[Fix] check to make sure processor has chat templates (#18047)
Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
This commit is contained in:
@@ -349,11 +349,11 @@ def resolve_hf_chat_template(
|
||||
trust_remote_code=model_config.trust_remote_code,
|
||||
)
|
||||
if isinstance(processor, ProcessorMixin) and \
|
||||
hasattr(processor, 'chat_template') and \
|
||||
processor.chat_template is not None:
|
||||
return processor.chat_template
|
||||
except Exception:
|
||||
logger.debug("Failed to load AutoProcessor chat template for %s",
|
||||
tokenizer.name_or_path, exc_info=True)
|
||||
logger.debug("Failed to load AutoProcessor chat template for %s", tokenizer.name_or_path, exc_info=True) # noqa: E501
|
||||
|
||||
# 3rd priority: AutoTokenizer chat template
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user