ssmits
/

Falcon2-5.5B-multilingual-embed-base

Text Classification

sentence-transformers

ssmits/Falcon2-5.5B-multilingual

Model card Files Files and versions

ssmits commited on Jun 10, 2024

Commit

cda5b4c

·

verified ·

1 Parent(s): 3fae013

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -97,7 +97,7 @@ print(sentence_embeddings)
 from transformers import AutoModel
 from torch.nn import DataParallel
-model = AutoModel.from_pretrained("ssmits/Qwen2-7B-embed-base")
 for module_key, module in model._modules.items():
     model._modules[module_key] = DataParallel(module)
 ```

 from transformers import AutoModel
 from torch.nn import DataParallel
+model = AutoModel.from_pretrained("ssmits/Falcon2-5.5B-multilingual-embed-base")
 for module_key, module in model._modules.items():
     model._modules[module_key] = DataParallel(module)
 ```