Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,8 @@ venv/
ENV/
env.bak/
venv.bak/
Scripts/
pyvenv.cfg

# Spyder project settings
.spyderproject
Expand Down
19 changes: 19 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,12 @@
# PyTorch CUDA Index - must be first line
--extra-index-url https://download.pytorch.org/whl/cu121

# PyTorch with CUDA support
torch==2.5.1
torchvision==0.20.1
torchaudio==2.5.1

# Dependencies
accelerate==1.6.0
diffusers==0.33.1
transformers==4.46.2
Expand All @@ -13,3 +22,13 @@ torchsde==0.2.6
einops
opencv-contrib-python
safetensors

# Video encoding support
ffmpeg-python>=0.2.0
imageio[ffmpeg]>=2.0.0
moviepy>=1.0.3


# Better Hugging Face performance
hf-transfer # Faster model downloads via Rust backend
# huggingface_hub[hf_xet] # Alternative: hf_xet for better download performance