Skip to content

Commit

Permalink
style
Browse files Browse the repository at this point in the history
  • Loading branch information
fxmarty committed Dec 12, 2023
1 parent c767d3e commit 6890d4c
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 @@ -1564,7 +1564,7 @@ def infer_library_from_model(
if not full_model_path.is_dir():
model_info = huggingface_hub.model_info(model_name_or_path, revision=revision)
library_name = getattr(model_info, "library_name", None)

# sentence-transformers package name is sentence_transformers
library_name = library_name.replace("-", "_")

Expand Down

0 comments on commit 6890d4c

Please sign in to comment.