Avoid using torch's Tensor or PIL's Image in chat template utils if n…

…ot available (huggingface#34165) * fix(utils): Avoid using torch Tensor or PIL Image if not available * Trigger CI --------- Co-authored-by: Matt <[email protected]>
byi8220 · Oct 16, 2024 · 3a10c61 · 3a10c61
1 parent bd5dc10
commit 3a10c61
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/transformers/utils/chat_template_utils.py b/src/transformers/utils/chat_template_utils.py
@@ -76,10 +76,12 @@ def _get_json_schema_type(param_type: str) -> Dict[str, str]:
         float: {"type": "number"},
         str: {"type": "string"},
         bool: {"type": "boolean"},
-        Image: {"type": "image"},
-        Tensor: {"type": "audio"},
         Any: {},
     }
+    if is_vision_available():
+        type_mapping[Image] = {"type": "image"}
+    if is_torch_available():
+        type_mapping[Tensor] = {"type": "audio"}
     return type_mapping.get(param_type, {"type": "object"})