Sign In

Hi-Dream installation Guide

18
Hi-Dream installation Guide

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:

https://developer.nvidia.com/cuda-12-4-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

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
18

Comments