UPDATE: SOME LATEST COMMIT FROM THIS NODE BREAK THIS INSTALLATION PROCESS. THIS PROCESS IS GOOD WITH COMMIT 90926d6ee290ae572f876246dd1b3fd8bdcbe826
Tested with 2.4 version of Cuda. It's probably good with 2.6 and 2.8 but I haven't tested.
This guide in for Windows.
NO NEED TO DOWNLOAD THE MODEL THE NODE WILL DO THAT FOR YOU
✅ CUDA Installation
Check CUDA version open the command prompt:
nvcc --version
Should be at least CUDA 12.4. If not, download and install:
Install Visual C++ Redistributable:
https://aka.ms/vs/17/release/vc_redist.x64.exe
Reboot you PC!!
✅ Triton Installation
Open command prompt:
pip uninstall triton-windows
pip install -U triton-windows
✅ Flash Attention Setup
Open command prompt:
Check Python version:
python --version
(3.10 and 3.11 are supported)
Check PyTorch version:
python
import torch
print(torch.__version__)
exit()
If the version is not 2.6.0+cu124:
pip uninstall torch torchvision torchaudio
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
Flash attention Wheel For Cuda 2.4 and python 3.10 Install:
If you use another version of Cuda than 2.4 of python version other than 3.10 go grab the right wheel link there:
https://huggingface.co/lldacing/flash-attention-windows-wheel/tree/main
Once you have the right wheel link use this command
pip install https://huggingface.co/lldacing/flash-attention-windows-wheel/resolve/main/flash_attn-2.7.4%2Bcu124torch2.6.0cxx11abiFALSE-cp310-cp310-win_amd64.whl
✅ ComfyUI + Nodes Installation
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI
pip install -r requirements.txt
Then go to custom_nodes folder and install the Node Manager and HiDream Sampler Node manually.
git clone https://github.com/Comfy-Org/ComfyUI-Manager.git
git clone https://github.com/lum3on/comfyui_HiDream-Sampler.git
get in the comfyui_HiDream-Sampler folder and run:
pip install -r requirements.txt
After that, type:
python -m pip install --upgrade transformers accelerate auto-gptq
Go back to the ComfyUi root folder
python main.py
A workflow should be in ComfyUI\custom_nodes\comfyui_HiDream-Sampler\sample_workflow
Edit:
Some people might have issue with tensor tensorflow. If it's your case use those commands.
pip uninstall tensorflow tensorflow-cpu tensorflow-gpu tf-nightly tensorboard Keras Keras-Preprocessing
pip install tensorflow