add Phi3MoE

xadupre · xadupre · commit 5a8eccfcc47a · 2025-06-21T11:25:05.000+02:00
diff --git a/onnx_diagnostic/torch_models/hghub/hub_api.py b/onnx_diagnostic/torch_models/hghub/hub_api.py
@@ -211,9 +211,7 @@ def task_from_id(
             f"The task can be added in "
             f"``onnx_diagnostic.torch_models.hghub.hub_data.__data_arch__``."
         )
-        return task_from_arch(
-            config.architectures[0], default_value=default_value, model_id=model_id
-        )
+        return task_from_arch(config.architectures[0], default_value=default_value)
 
 
 def task_from_tags(tags: Union[str, List[str]]) -> str:
diff --git a/onnx_diagnostic/torch_models/hghub/hub_data.py b/onnx_diagnostic/torch_models/hghub/hub_data.py
@@ -99,6 +99,7 @@
     PegasusModel,feature-extraction
     Phi3ForCausalLM,text-generation
     PhiForCausalLM,text-generation
+    PhiMoEForCausalLM,text-generation
     Pix2StructForConditionalGeneration,image-to-text
     PLBartForConditionalGeneration,text2text-generation
     PoolFormerModel,image-feature-extraction

Original file line number	Diff line number	Diff line change
`@@ -211,9 +211,7 @@ def task_from_id(`
`211`	`211`	`f"The task can be added in "`
`212`	`212`	f"``onnx_diagnostic.torch_models.hghub.hub_data.__data_arch__``."
`213`	`213`	`)`
`214`		`- return task_from_arch(`
`215`		`- config.architectures[0], default_value=default_value, model_id=model_id`
`216`		`- )`
	`214`	`+ return task_from_arch(config.architectures[0], default_value=default_value)`
`217`	`215`
`218`	`216`
`219`	`217`	`def task_from_tags(tags: Union[str, List[str]]) -> str:`