diff --git a/docs/attention/index.md b/docs/attention/index.md index f77c7b742..b24098c87 100644 --- a/docs/attention/index.md +++ b/docs/attention/index.md @@ -41,7 +41,7 @@ Clone the repository and build the kernel: ```bash # Clone recursively to get ThunderKittens submodule -git clone --recursive https://github.com/hao-ai-lab/FastVideo.git +git clone https://github.com/hao-ai-lab/FastVideo.git cd FastVideo/fastvideo-kernel # Build and install diff --git a/docs/getting_started/installation.md b/docs/getting_started/installation.md index c02abc332..99f1eb908 100644 --- a/docs/getting_started/installation.md +++ b/docs/getting_started/installation.md @@ -11,15 +11,13 @@ FastVideo supports the following hardware platforms: ### Using pip ```bash +# Create and activate a new conda environment +conda create -n fastvideo python=3.12 +conda activate fastvideo + pip install fastvideo ``` -### Using conda - -```bash -conda install -c conda-forge fastvideo -``` - ### From source ```bash @@ -28,6 +26,12 @@ cd FastVideo pip install -e . ``` +Also optionally install flash-attn: + +```bash +pip install flash-attn --no-build-isolation +``` + ## Hardware Requirements - **NVIDIA GPUs**: CUDA 11.8+ with compute capability 7.0+ diff --git a/docs/getting_started/quick_start.md b/docs/getting_started/quick_start.md index a2fd39168..1cb56df52 100644 --- a/docs/getting_started/quick_start.md +++ b/docs/getting_started/quick_start.md @@ -15,6 +15,12 @@ conda activate fastvideo pip install fastvideo ``` +Also optionally install flash-attn: + +```bash +pip install flash-attn --no-build-isolation +``` + ## Basic Usage ### Text-to-Video Generation diff --git a/pyproject.toml b/pyproject.toml index af320eb18..99d5bcece 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -63,6 +63,7 @@ dependencies = [ "remote-pdb", # Kernel & Packaging + "fastvideo-kernel==0.2.2", "wheel", # Training Dependencies diff --git a/pyproject_other.toml b/pyproject_other.toml index 671d2a0eb..e97043d0e 100644 --- a/pyproject_other.toml +++ b/pyproject_other.toml @@ -63,6 +63,7 @@ dependencies = [ "remote-pdb", # Kernel & Packaging + "fastvideo-kernel==0.2.2", "wheel", # Training Dependencies