maxdiffusion/requirements.txt at e5a9df2cd3b06a1e2d447bea4007ab0b9b38e1b0 · AI-Hypercomputer/maxdiffusion · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
--extra-index-url https://download.pytorch.org/whl/cpu
jax>=0.7.2
jaxlib>=0.4.30
grain
google-cloud-storage>=2.17.0
absl-py
datasets
flax>=0.12.0
optax>=0.2.3
torch>=2.6.0
torchvision>=0.20.1
ftfy
tensorboard>=2.17.0
tensorboardx>=2.6.2.2
tensorboard-plugin-profile>=2.15.2
tokamax
Jinja2
scikit-image
parameterized
Pillow
pylint
pyink
pytest==8.2.2
tensorflow>=2.17.0
tensorflow-datasets>=4.9.6
ruff>=0.1.5,<=0.2
git+https://github.com/Lightricks/LTX-Video.git#egg=ltx-video[inference]
git+https://github.com/zmelumian972/xla@torchax/jittable_module_callable#subdirectory=torchax
opencv-python-headless==4.10.0.84
orbax-checkpoint
tokenizers==0.21.0
huggingface_hub>=0.30.2
transformers==4.48.1
einops==0.8.0
sentencepiece
aqtp
imageio==2.37.0
imageio-ffmpeg==0.6.0
hf_transfer>=0.1.9
qwix@git+https://github.com/google/qwix.git