diff --git a/README.md b/README.md index 8409827..83297e8 100644 --- a/README.md +++ b/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/flash-attention.git@v2.1.2.post3 +``` + ## 🧱 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).) diff --git a/deepspeed-0.6.3-py3-none-any.whl b/deepspeed-0.6.3-py3-none-any.whl new file mode 100644 index 0000000..42fca79 Binary files /dev/null and b/deepspeed-0.6.3-py3-none-any.whl differ diff --git a/diffusers-0.21.2-py3-none-any.whl b/diffusers-0.21.2-py3-none-any.whl new file mode 100644 index 0000000..819461f Binary files /dev/null and b/diffusers-0.21.2-py3-none-any.whl differ diff --git a/environment.yml b/environment.yml index f43b45b..ab26610 100644 --- a/environment.yml +++ b/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 diff --git a/nvidia_pyindex-1.0.9-py3-none-any.whl b/nvidia_pyindex-1.0.9-py3-none-any.whl new file mode 100644 index 0000000..946a1e0 Binary files /dev/null and b/nvidia_pyindex-1.0.9-py3-none-any.whl differ