revert requirements to original.

Author: jfacevedo-google

requirements.txt

@@ -1,171 +1,39 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-absl-py==2.3.1
-aiofiles==24.1.0
-aiohappyeyeballs==2.6.1
-aiohttp==3.12.15
-aiosignal==1.4.0
-annotated-types==0.7.0
-aqtp==0.9.0
-array_record==0.8.1
-astroid==3.3.11
-astunparse==1.6.3
-attrs==25.3.0
-black==24.10.0
-cachetools==5.5.2
-certifi==2025.8.3
-charset-normalizer==3.4.3
-cheroot==11.0.0
-chex==0.1.91
-click==8.3.0
-cloudpickle==3.1.1
-datasets==4.1.1
-decorator==5.2.1
-diffusers==0.35.1
-dill==0.4.0
-dm-tree==0.1.9
-docstring_parser==0.17.0
-einops==0.8.0
-etils==1.13.0
-filelock==3.19.1
-flatbuffers==25.9.23
-flax==0.12.0
-frozenlist==1.7.0
-fsspec==2025.9.0
-ftfy==6.3.1
-gast==0.6.0
-gcsfs==2025.9.0
-google-api-core==2.25.1
-google-auth==2.40.3
-google-auth-oauthlib==1.2.2
-google-cloud-core==2.4.3
-google-cloud-storage==3.4.0
-google-crc32c==1.7.1
-google-pasta==0.2.0
-google-resumable-media==2.7.2
-googleapis-common-protos==1.70.0
-grain==0.2.12
-grpcio==1.75.1
-gviz-api==1.10.0
-h5py==3.14.0
-hf-xet==1.1.10
-hf_transfer==0.1.9
-huggingface-hub==0.35.1
-humanize==4.13.0
-idna==3.10
-imageio==2.37.0
-imageio-ffmpeg==0.6.0
-immutabledict==4.2.1
-importlib_metadata==8.7.0
-importlib_resources==6.5.2
-iniconfig==2.1.0
-isort==6.0.1
-jaraco.functools==4.3.0
-jax==0.7.2
-jax-cuda13-pjrt==0.7.2
-jax-cuda13-plugin==0.7.2
-jaxlib==0.7.2
-Jinja2==3.1.6
-keras==3.11.3
-lazy_loader==0.4
-libclang==18.1.1
-ltx-video @ git+https://github.com/Lightricks/LTX-Video@53d263f31727a0021bf65e3e413d1cb139abb86b
-Markdown==3.9
-markdown-it-py==4.0.0
-MarkupSafe==3.0.2
-maxdiffusion @ file:///app
-mccabe==0.7.0
-mdurl==0.1.2
-ml_dtypes==0.5.3
-more-itertools==10.8.0
-mpmath==1.3.0
-msgpack==1.1.1
-multidict==6.6.4
-multiprocess==0.70.16
-mypy_extensions==1.1.0
-namex==0.1.0
-nest-asyncio==1.6.0
-networkx==3.5
-numpy==2.3.3
-oauthlib==3.3.1
-opencv-python-headless==4.10.0.84
-opt_einsum==3.4.0
-optax==0.2.6
-optree==0.17.0
-orbax-checkpoint==0.11.25
-packaging==25.0
-pandas==2.3.2
-parameterized==0.9.0
-pathspec==0.12.1
-pillow==11.3.0
-platformdirs==4.4.0
-pluggy==1.6.0
-promise==2.3
-propcache==0.3.2
-proto-plus==1.26.1
-protobuf==6.32.1
-psutil==7.1.0
-pyarrow==21.0.0
-pyasn1==0.6.1
-pyasn1_modules==0.4.2
-pybind11==3.0.1
-pydantic==2.11.9
-pydantic_core==2.33.2
-Pygments==2.19.2
-pyink==24.10.1
-pylint==3.3.8
+jax>=0.6.2
+jaxlib>=0.4.30
+grain
+google-cloud-storage>=2.17.0
+absl-py
+datasets
+flax>=0.11.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
+Jinja2
+scikit-image
+parameterized
+Pillow
+pylint
+pyink
 pytest==8.2.2
-python-dateutil==2.9.0.post0
-pytz==2025.2
-PyYAML==6.0.3
-qwix @ git+https://github.com/google/qwix.git@e99e27f517ec8de00c29dd39c638e44e14a215fd
-regex==2025.9.18
-requests==2.32.5
-requests-oauthlib==2.0.0
-rich==14.1.0
-rsa==4.9.1
-ruff==0.2.0
-safetensors==0.6.2
-scikit-image==0.25.2
-scipy==1.16.2
-sentencepiece==0.2.1
-setuptools==68.1.2
-simple-parsing==0.1.7
-simplejson==3.20.2
-six==1.17.0
-sympy==1.14.0
-tensorboard==2.20.0
-tensorboard-data-server==0.7.2
-tensorboard_plugin_profile==2.20.6
-tensorboardX==2.6.4
-tensorflow==2.20.0
-tensorflow-datasets==4.9.9
-tensorflow-metadata==1.17.2
-tensorstore==0.1.77
-termcolor==3.1.0
-tifffile==2025.9.20
-timm==1.0.20
+tensorflow>=2.17.0
+tensorflow-datasets>=4.9.6
+ruff>=0.1.5,<=0.2
+git+https://github.com/Lightricks/LTX-Video
+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
-toml==0.10.2
-tomlkit==0.13.3
-toolz==1.0.0
-torch==2.8.0+cpu
-torchax @ git+https://github.com/zmelumian972/xla@f1f6894bb63bfd1d30893f6da500fbfced72b5cd#subdirectory=torchax
-torchvision==0.23.0
-tqdm==4.67.1
-transformer-engine-jax==2.6.0
-transformer_engine==2.6.0
-transformer_engine_cu12==2.6.0
+huggingface_hub>=0.30.2
 transformers==4.48.1
-treescope==0.1.10
-typing-inspection==0.4.1
-typing_extensions==4.15.0
-tzdata==2025.2
-urllib3==2.5.0
-wcwidth==0.2.14
-Werkzeug==3.1.3
-wheel==0.42.0
-wrapt==1.17.3
-xprof==2.20.6
-xxhash==3.5.0
-yarl==1.20.1
-zipp==3.23.0
+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