zero

Author: xadupre

CHANGELOGS.rst

@@ -4,6 +4,7 @@ Change Logs
 0.4.0
 +++++
 
+* :pr:`52`: add support for zero-shot-image-classification
 * :pr:`50`: add support for onnxruntime fusion
 * :pr:`48`: add support for EncoderDecoderCache, test with openai/whisper-tiny
 * :pr:`45`: improve change_dynamic_dimension to fix some dimensions

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -279,9 +279,10 @@ def random_input_kwargs(config: Any, task: str) -> Tuple[Dict[str, Any], Callabl
             )
         fct = get_inputs_for_image_classification  # type: ignore
     elif task == "zero-shot-image-classification":
-        check_hasattr(config, "vision_config", "text_config")
-        check_hasattr(config.vision_config, "image_size", "num_channels")
-        check_hasattr(config.text_config, "vocab_size")
+        if config is not None:
+            check_hasattr(config, "vision_config", "text_config")
+            check_hasattr(config.vision_config, "image_size", "num_channels")
+            check_hasattr(config.text_config, "vocab_size")
         kwargs = dict(
             batch_size=2,
             batch_size_image=3,