Former-commit-id: d41d43a7c37cd10e34c9f399d1a346ffaee641c3
This commit is contained in:
hiyouga 2024-09-04 03:06:06 +08:00
parent a65b5061af
commit a3d47818b7

View File

@ -349,7 +349,7 @@ def get_template_and_fix_tokenizer(
name: Optional[str] = None, name: Optional[str] = None,
tool_format: Optional[str] = None, tool_format: Optional[str] = None,
) -> Template: ) -> Template:
if name == "qwen2_vl": if name in ["llava", "paligemma", "qwen2_vl"]:
require_version( require_version(
"transformers>=4.45.0.dev0", "To fix: pip install git+https://github.com/huggingface/transformers.git" "transformers>=4.45.0.dev0", "To fix: pip install git+https://github.com/huggingface/transformers.git"
) )