Skip to content

Commit

Permalink
Update optimum/exporters/tasks.py
Browse files Browse the repository at this point in the history
  • Loading branch information
fxmarty authored Oct 16, 2023
1 parent 89cbd3f commit a253ca1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/exporters/tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -1401,7 +1401,7 @@ def _infer_task_from_model_name_or_path(
pipeline_tag = getattr(model_info, "pipeline_tag", None)
# The Hub task "conversational" is not a supported task per se, just an alias that may map to
# text-generaton or text2text-generation.
# The Hub task "object-detection" is not a supported task per so, as in Transformers this may map to either
# The Hub task "object-detection" is not a supported task per se, as in Transformers this may map to either
# zero-shot-object-detection or object-detection.
if pipeline_tag is not None and pipeline_tag not in ["conversational", "object-detection"]:
inferred_task_name = TasksManager.map_from_synonym(model_info.pipeline_tag)
Expand Down

0 comments on commit a253ca1

Please sign in to comment.