Pass the revision to SentenceTransformer models (#2105)

feat: pass revision to SentenceTransformers
huggingface · Nov 28, 2024 · 28bd0ad · 28bd0ad
1 parent bd08f12
commit 28bd0ad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
@@ -2128,6 +2128,7 @@ def get_model_from_task(
                 device=device,
                 cache_folder=cache_folder,
                 token=token,
+                revision=revision,
                 trust_remote_code=trust_remote_code,
             )
         else: