Wan2.1/INSTALL.md
gnebie 0f816057da
Add a pyproject and a separate INSTALL.md (#137)
* Add a pyproject and a detailled separate installation file with a poetry installation to not overcharge the readme

* update mail
2025-03-20 10:16:23 +08:00

1.1 KiB

Installation Guide

Install with pip

pip install .
pip install .[dev]  # Installe aussi les outils de dev

Install with Poetry

Ensure you have Poetry installed on your system.

To install all dependencies:

poetry install

Handling flash-attn Installation Issues

If flash-attn fails due to PEP 517 build issues, you can try one of the following fixes.

poetry run pip install --upgrade pip setuptools wheel
poetry run pip install flash-attn --no-build-isolation
poetry install

Install from Git (Alternative)

poetry run pip install git+https://github.com/Dao-AILab/flash-attention.git

Running the Model

Once the installation is complete, you can run Wan2.1 using:

poetry run python generate.py --task t2v-14B --size '1280x720' --ckpt_dir ./Wan2.1-T2V-14B --prompt "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."

Test

pytest tests/

Format

black .
isort .