ssmits
/

Qwen2-7B-embed-base

Text Classification

sentence-transformers

Model card Files Files and versions Community

ssmits commited on Jun 10

Commit

963927d

•

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