Hugging Face's logo

X-Humanoid
/
WoW-1-DiT-2B-600k

Model card Files
xet
 Community

YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

This is a README for environment setup and inference, not for training so far.

Installation Setup 

conda create -n wow-dit-2b python=3.10 -y
pip install transformer-engine[pytorch]==1.12.0
git clone https://github.com/NVIDIA/apex
CUDA_HOME=$CONDA_PREFIX pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./apex
pip install -r dit_models/wow-dit-2b/requires.txt

Inference 

cd scripts
conda activate wow-dit-2b
python infer_wow_dit_2b.py

Installation Issues

  • Conda environment conflicts: Create fresh environment with conda create -n wow-dit-2b-clean python=3.10 -y
  • Flash-attention build failures: Install build tools with apt-get install build-essential
  • Transformer engine linking errors: Reinstall with pip install --force-reinstall transformer-engine==1.12.0
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Collection including X-Humanoid/WoW-1-DiT-2B-600k