Downloads · 30 days
0
WoW-1-DiT-2B-600k
X-Humanoid/WoW-1-DiT-2B-600k
WoW-1-DiT-2B-600k is a machine learning model from X-Humanoid. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This is a README for environment setup and inference, not for training so far.
Access
Public
Updated Oct 16, 2025
Repo size
33.4 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.mp433.4 MB · 85%
At a glance
- Access
- Public
- Created
- Oct 15, 2025
- Updated
- Oct 16, 2025
- SHA
- 230b975e
Model card
From the Hugging Face model README
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
- Created
- Oct 15, 2025
- Updated
- Oct 16, 2025