fix issues

Author: xadupre

_unittests/ut_torch_models/test_hghub_model.py

@@ -49,7 +49,11 @@ def test_get_untrained_model_with_inputs_phi_2(self):
         data = get_untrained_model_with_inputs(mid, verbose=1)
         model, inputs = data["model"], data["inputs"]
         model(**inputs)
-        self.assertEqual((1040498688, 260124672), (data["size"], data["n_weights"]))
+        # different expected value for different version of transformers
+        self.assertIn(
+            (data["size"], data["n_weights"]),
+            [(1040293888, 260073472), (1040498688, 260124672)],
+        )
 
 
 if __name__ == "__main__":

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -53,9 +53,9 @@ def get_untrained_model_with_inputs(
     Gets a non initialized model similar to the original model
     based on the model id given to the function.
     The model size is reduced compare to the original model.
-    No weights download is required.
+    No weight is downloaded, only the configuration file sometimes.
 
-    :param model_id: model id, ex:`arnir0/Tiny-LLM`
+    :param model_id: model id, ex: :epkg:`arnir0/Tiny-LLM`
     :param config: to overwrite the configuration
     :param task: model task, can be overwritten, otherwise, it is automatically determined
     :param input_kwargs: parameters sent to input generation