ssmits
/

Qwen2-7B-embed-base

Feature Extraction

text-generation-inference

text-embeddings-inference

Model card Files Files and versions

ssmits commited on Jun 10, 2024

Commit

963927d

·

verified ·

1 Parent(s): a31126a

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

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

 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)
 ```