quotes

Author: xadupre

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -318,11 +318,11 @@ def get_untrained_model_with_inputs(
 
         data = get_untrained_model_with_inputs("arnir0/Tiny-LLM", verbose=1)
 
-        print("-- model size:", data["size"])
-        print("-- number of parameters:", data["n_weights"])
-        print("-- inputs:", string_type(data["inputs"], with_shape=True))
-        print("-- dynamic shapes:", pprint.pformat(data["dynamic_shapes"]))
-        print("-- configuration:", pprint.pformat(data["configuration"]))
+        print("-- model size:", data['size'])
+        print("-- number of parameters:", data['n_weights'])
+        print("-- inputs:", string_type(data['inputs'], with_shape=True))
+        print("-- dynamic shapes:", pprint.pformat(data['dynamic_shapes']))
+        print("-- configuration:", pprint.pformat(data['configuration']))
     """
     if verbose:
         print(f"[get_untrained_model_with_inputs] model_id={model_id!r}")

onnx_diagnostic/torch_models/test_helper.py

@@ -181,7 +181,7 @@ def validate_model(
     if drop_inputs:
         if verbose:
             print(f"[validate_model] drop inputs {drop_inputs!r}")
-            print(f"[validate_model] current inputs: {string_type(data["inputs"])}")
+            print(f"[validate_model] current inputs: {string_type(data['inputs'])}")
             print(
                 f"[validate_model] current dynnamic_shapes: "
                 f"{_ds_clean(data["dynamic_shapes"])}"