Skip to content

Commit cb4ac9b

Browse files
tastelikefeettastelikefeet
andauthored
fix qwen3 embedding saving (#4548)
--------- Co-authored-by: tastelikefeet <[email protected]>
1 parent 89fe048 commit cb4ac9b

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

swift/trainers/mixin.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,11 @@ def _save_model(self, output_dir: Optional[str] = None, state_dict=None):
223223
else:
224224
if self.model.__class__.__name__ != 'SentenceTransformer':
225225
self.model.save_pretrained(output_dir, state_dict=state_dict, safe_serialization=save_safetensors)
226+
# For embedding models, they should copy extra sentence_transformers files
227+
from swift.utils import copy_files_by_pattern
228+
copy_files_by_pattern(self.model.model_dir, output_dir, 'config_sentence_transformers.json')
229+
copy_files_by_pattern(self.model.model_dir, output_dir, '1_Pooling/config.json')
230+
copy_files_by_pattern(self.model.model_dir, output_dir, 'modules.json')
226231
else:
227232

228233
@contextmanager

0 commit comments

Comments
 (0)