fix llm

Author: xadupre

_unittests/ut_torch_models/test_hghub_api.py

@@ -44,6 +44,16 @@ def test_task_from_id(self):
         for name, etask in [
             ("arnir0/Tiny-LLM", "text-generation"),
             ("microsoft/phi-2", "text-generation"),
+        ]:
+            with self.subTest(name=name, task=etask):
+                task = task_from_id(name, True)
+                self.assertEqual(etask, task)
+
+    @requires_transformers("4.50")
+    @requires_torch("2.7")
+    @never_test()
+    def test_task_from_id_long(self):
+        for name, etask in [
             ("microsoft/Phi-3.5-mini-instruct", "text-generation"),
             ("microsoft/Phi-3.5-vision-instruct", "text-generation"),
         ]:

onnx_diagnostic/torch_models/hghub/model_inputs.py

@@ -10,7 +10,7 @@
 
 
 @functools.cache
-def config_class_from_architecture(arch: str, exc: bool = False) -> type:
+def config_class_from_architecture(arch: str, exc: bool = False) -> Optional[type]:
     """
     Retrieves the configuration class for a given architecture.