ssmits commited on
Commit
cda5b4c
1 Parent(s): 3fae013

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -97,7 +97,7 @@ print(sentence_embeddings)
97
  from transformers import AutoModel
98
  from torch.nn import DataParallel
99
 
100
- model = AutoModel.from_pretrained("ssmits/Qwen2-7B-embed-base")
101
  for module_key, module in model._modules.items():
102
  model._modules[module_key] = DataParallel(module)
103
  ```
 
97
  from transformers import AutoModel
98
  from torch.nn import DataParallel
99
 
100
+ model = AutoModel.from_pretrained("ssmits/Falcon2-5.5B-multilingual-embed-base")
101
  for module_key, module in model._modules.items():
102
  model._modules[module_key] = DataParallel(module)
103
  ```