From 27eea36b0928bb8aa3b3d7703f52121eac99903f Mon Sep 17 00:00:00 2001 From: Brando Tovar Date: Tue, 26 Nov 2024 22:53:54 -0500 Subject: [PATCH] feat: pass revision to SentenceTransformers --- optimum/exporters/tasks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index 8f28ec42ce9..c50fa5cdfa4 100644 --- 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: