From f60abc289d2e78d2148755b679351ce1bde98730 Mon Sep 17 00:00:00 2001
From: chuxij <junmingong@qq.com>
Date: Mon, 5 May 2025 03:54:44 +0000
Subject: [PATCH] update win install support

---
 README.md | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/README.md b/README.md
index 69a0691..cae09e4 100644
--- a/README.md
+++ b/README.md
@@ -115,10 +115,33 @@ Rather than building yet another end-to-end text-to-music pipeline, our vision i
 
 ## 💻 Installation
 
+### For MacOS/Linux
+
 ```bash
 conda create -n ace_step python==3.10
 conda activate ace_step
+# Install other requirements
 pip install -r requirements.txt
+
+# Install ffmpeg
+conda install ffmpeg
+```
+
+### Windows
+
+```bash
+conda create -n ace_step python==3.10
+conda activate ace_step
+
+# Install PyTorch, TorchAudio, and TorchVision for Windows
+# replace cu121 with your CUDA version
+# replace torchvision and torchaudio with your version
+pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
+
+# Install other requirements
+pip install -r requirements_win.txt
+
+# Install ffmpeg
 conda install ffmpeg
 ```