This commit is contained in:
hiyouga
2024-09-04 03:06:06 +08:00
parent ce7ed6eb99
commit d41d43a7c3

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"
) )