1.add deepspeed,diffusers and nvidia-pyindex whl file for pip install…

README.md

@@ -288,6 +288,30 @@ docker image ls
 docker run -dit --gpus all --init --net=host --uts=host --ipc=host --name hunyuandit --security-opt=seccomp=unconfined --ulimit=stack=67108864 --ulimit=memlock=-1 --privileged  docker_image_tag
 ```
 
+### Installation Guide for windows
+build Windows environments with python=3.10 and torch=2.1.2 cuda=12.1 for mathing comfyUI
+```shell
+# 1. Prepare conda environment
+conda  create -n HunyuanDiT python==3.10.11
+
+# 2. Activate the environment
+conda activate HunyuanDiT
+
+# 3. Install pytorch
+pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
+
+# 4. Install deepspeed diffusers and nvidia-pyindex
+pip install deepspeed-0.6.3-py3-none-any.whl
+pip install diffusers-0.21.2-py3-none-any.whl
+pip install nvidia_pyindex-1.0.9-py3-none-any.whl
+
+# 5. Install pip dependencies
+pip install -r requirements.txt
+
+# 6. Install flash attention v2 for acceleration (requires CUDA 11.6 or above)
+pip install git+https://github.com/Dao-AILab/[email protected]
+```
+
 ## 🧱 Download Pretrained Models
 To download the model, first install the huggingface-cli. (Detailed instructions are available [here](https://huggingface.co/docs/huggingface_hub/guides/cli).)
 

environment.yml

@@ -3,6 +3,6 @@ channels:
   - pytorch
   - nvidia
 dependencies:
-  - python=3.8.12
-  - pytorch=1.13.1
+  - python=3.10.11
+  - pytorch=2.1.2+cu121
   - pip