Update conversion script

xenova · xenova · commit d7438ce26dc6 · 2023-09-10T00:44:00.000+02:00
Use latest version of Optimum (source)
diff --git a/scripts/convert.py b/scripts/convert.py
@@ -114,7 +114,7 @@ class ConversionArguments:
         metadata={
             "help": (
                 "The task to export the model for. If not specified, the task will be auto-inferred based on the model. Available tasks depend on the model, but are among:"
-                f" {str(list(TasksManager._TASKS_TO_AUTOMODELS.keys()))}. For decoder models, use `xxx-with-past` to export the model using past key values in the decoder."
+                f" {str(TasksManager.get_all_tasks())}. For decoder models, use `xxx-with-past` to export the model using past key values in the decoder."
             )
         }
     )

Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ class ConversionArguments:`
`114`	`114`	`metadata={`
`115`	`115`	`"help": (`
`116`	`116`	`"The task to export the model for. If not specified, the task will be auto-inferred based on the model. Available tasks depend on the model, but are among:"`
`117`		- f" {str(list(TasksManager._TASKS_TO_AUTOMODELS.keys()))}. For decoder models, use `xxx-with-past` to export the model using past key values in the decoder."
	`117`	+ f" {str(TasksManager.get_all_tasks())}. For decoder models, use `xxx-with-past` to export the model using past key values in the decoder."
`118`	`118`	`)`
`119`	`119`	`}`
`120`	`120`	`)`