add mpi4py and deepspeed install

intel-analytics · Jul 1, 2024 · d1434bf · d1434bf
1 parent e088926
commit d1434bf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/llm/example/GPU/LLM-Finetuning/LoRA/chatglm_finetune/README.md b/python/llm/example/GPU/LLM-Finetuning/LoRA/chatglm_finetune/README.md
@@ -17,6 +17,8 @@ pip install typer
 pip install sentencepiece
 pip install nltk
 pip install "numpy<2.0.0"
+pip install "deepspeed==0.13.1"
+pip install "mpi4py>=3.1.5"
 # below command will install intel_extension_for_pytorch==2.1.10+xpu as default
 pip install --pre --upgrade ipex-llm[xpu] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
 pip install oneccl_bind_pt==2.1.100 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/