Update README.md

README.md

@@ -91,4 +91,14 @@ sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask']
 
 print("Sentence embeddings:")
 print(sentence_embeddings)
+```
+
+### How to enable Multi-GPU
+```python
+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)
 ```