fix config

Author: xadupre

onnx_diagnostic/tasks/image_text_to_text.py

@@ -167,27 +167,26 @@ def random_input_kwargs(config: Any) -> Tuple[Dict[str, Any], Callable]:
             else getattr(
                 config,
                 "head_dim",
-                (config.text_config.hidden_size if text_config else config.config.hidden_size)
+                (config.text_config.hidden_size if text_config else config.hidden_size)
                 // (
                     config.text_config.num_attention_heads
                     if text_config
-                    else config.config.num_attention_heads
+                    else config.num_attention_heads
                 ),
             )
         ),
         dummy_max_token_id=(
             31999
             if config is None
-            else (config.text_config.vocab_size if text_config else config.config.vocab_size)
-            - 1
+            else (config.text_config.vocab_size if text_config else config.vocab_size) - 1
         ),
         num_hidden_layers=(
             4
             if config is None
             else (
                 config.text_config.num_hidden_layers
                 if text_config
-                else config.config.num_hidden_layers
+                else config.num_hidden_layers
             )
         ),
         num_key_value_heads=(
@@ -205,7 +204,7 @@ def random_input_kwargs(config: Any) -> Tuple[Dict[str, Any], Callable]:
             else (
                 config.text_config.intermediate_size
                 if text_config
-                else config.config.intermediate_size
+                else config.intermediate_size
             )
         ),
         hidden_size=(

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -72,7 +72,7 @@ def get_untrained_model_with_inputs(
         print("-- configuration:", pprint.pformat(data['configuration']))
     """
     assert not use_preinstalled or not use_only_preinstalled, (
-        f"model_id={model_id!r}, pretinstalled model is only avaialble "
+        f"model_id={model_id!r}, pretinstalled model is only available "
         f"if use_only_preinstalled is False."
     )
     if verbose: