From 6890d4c4c676d66eee38417004d9054bafefe7da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?F=C3=A9lix=20Marty?= <9808326+fxmarty@users.noreply.github.com> Date: Tue, 12 Dec 2023 16:20:10 +0100 Subject: [PATCH] style --- optimum/exporters/tasks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index b1add67a13..c1c37cdc0b 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -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("-", "_")