Don't call set_base_model when local_files_only=True

UKPLab · Jan 28, 2025 · cb929c4 · cb929c4
1 parent 3fbc183
commit cb929c4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py
@@ -1544,7 +1544,8 @@ def _load_auto_model(
             backend=self.backend,
         )
         pooling_model = Pooling(transformer_model.get_word_embedding_dimension(), "mean")
-        self.model_card_data.set_base_model(model_name_or_path, revision=revision)
+        if not local_files_only:
+            self.model_card_data.set_base_model(model_name_or_path, revision=revision)
         return [transformer_model, pooling_model]
 
     def _load_module_class_from_ref(
@@ -1762,7 +1763,8 @@ def _load_sbert_model(
                 revision_path_part = Path(modules_json_path).parts[-2]
                 if len(revision_path_part) == 40:
                     revision = revision_path_part
-        self.model_card_data.set_base_model(model_name_or_path, revision=revision)
+        if not local_files_only:
+            self.model_card_data.set_base_model(model_name_or_path, revision=revision)
         return modules, module_kwargs
 
     @staticmethod