install deepspeed only on linux

2026-02-05 18:09:24 +08:00 · 2024-07-07 16:45:30 +08:00
parent 39565cc02c
commit 89fc7220ea
2 changed files with 3 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
 conformer==0.3.2
-deepspeed==0.14.2
+deepspeed==0.14.2; sys_platform == 'linux'
 diffusers==0.27.2
 gdown==5.1.0
 gradio==4.32.2