返回 JoyAI-Echo
requirements.txt
根目录 / echo_wm / requirements.txt
1 # Core dependencies
2 # transformers is capped below 5.0: the bundled ltx-core text encoder expects the
3 # 4.x Siglip layout (vision_tower.vision_model), which 5.x restructured.
4 # torch is pinned to the exact version this release was validated on. Install it
5 # from the cu128 wheel index first (see README), then install this file; a bare
6 # PyPI torch can mismatch the CUDA runtime.
7 torch==2.9.1
8 torchaudio==2.9.1
9 torchvision==0.24.1
10 einops
11 numpy
12 transformers>=4.52,<5.0
13 safetensors
14 accelerate
15 scipy>=1.14
16
17 # Video processing
18 opencv-python
19 imageio
20 imageio-ffmpeg
21 av
22
23 # Optional acceleration.
24 # xformers builds track a specific torch release: 0.0.35 targets torch 2.11 and
25 # fails to import against the pinned 2.9.1. Keep this aligned with torch above.
26 xformers==0.0.33.post2
27
28 # UI and visualization
29 # gradio is capped below 6.0: 6.x requires huggingface-hub>=1.16, which conflicts
30 # with the huggingface-hub<1.0 that transformers 4.x requires.
31 gradio>=5.0,<6.0
32 matplotlib
33 pillow
34
35 # Utilities
36 tqdm
37 pyyaml
38 omegaconf
39
40 # Hugging Face integration
41 huggingface-hub<1.0
42
43 # Additional utilities
44 sentencepiece
45 protobuf
46
46 lines Plain Text