Update README.md
Browse files
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/
|
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 |
```
|