Pass the revision to SentenceTransformer models (#2105)

bndos · web-flow · commit 28bd0ad8fccf · 2024-11-28T16:13:05.000+01:00
feat: pass revision to SentenceTransformers
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:

Original file line number	Diff line number	Diff line change
`@@ -2128,6 +2128,7 @@ def get_model_from_task(`
`2128`	`2128`	`device=device,`
`2129`	`2129`	`cache_folder=cache_folder,`
`2130`	`2130`	`token=token,`
	`2131`	`+ revision=revision,`
`2131`	`2132`	`trust_remote_code=trust_remote_code,`
`2132`	`2133`	`)`
`2133`	`2134`	`else:`