feat: pass revision to SentenceTransformers

huggingface · Nov 27, 2024 · 27eea36 · 27eea36
1 parent a6c696c
commit 27eea36
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: