From 3b1682a43396eb8a5ecb0748049b6845390816e8 Mon Sep 17 00:00:00 2001 From: Rohan Bierneni Date: Tue, 29 Jul 2025 00:42:58 +0000 Subject: [PATCH] Updated requirements.txt for to test jaii deps --- requirements.txt | 217 +++++++++++++++++++++++++++++++++++++++-------- 1 file changed, 182 insertions(+), 35 deletions(-) diff --git a/requirements.txt b/requirements.txt index b66f5823..7f92be52 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,38 +1,185 @@ ---extra-index-url https://download.pytorch.org/whl/cpu -jax>=0.6.2 -jaxlib>=0.4.30 -grain -google-cloud-storage>=2.17.0 -absl-py -datasets -flax>=0.10.2 -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 -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 -huggingface_hub>=0.30.2 -transformers==4.48.1 +absl-py==2.3.1 +aiohappyeyeballs==2.6.1 +aiohttp==3.12.14 +aiosignal==1.4.0 +annotated-types==0.7.0 +anyio==4.9.0 +aqtp==0.8.4 +array_record==0.7.2 +astroid==3.3.11 +astunparse==1.6.3 +attrs==25.3.0 +black==24.10.0 +cachetools==5.5.2 +certifi==2025.7.14 +charset-normalizer==3.4.2 +cheroot==10.0.1 +chex==0.1.90 +click==8.2.1 +cloud-accelerator-diagnostics==0.1.1 +cloud-tpu-diagnostics==0.1.5 +cloudpickle==3.1.1 +clu==0.0.12 +datasets==4.0.0 +decorator==5.2.1 +dill==0.3.8 +dm-tree==0.1.9 +docstring_parser==0.17.0 einops==0.8.0 -sentencepiece -aqtp +etils==1.13.0 +fiddle==0.3.0 +filelock==3.18.0 +flatbuffers==25.2.10 +flax==0.10.6 +frozenlist==1.7.0 +fsspec==2025.3.0 +ftfy==6.3.1 +gast==0.6.0 +gcsfs==2025.3.0 +google-api-core==2.25.1 +google-auth==2.40.3 +google-auth-oauthlib==1.2.2 +google-cloud-aiplatform==1.105.0 +google-cloud-appengine-logging==1.6.2 +google-cloud-audit-log==0.3.2 +google-cloud-bigquery==3.35.1 +google-cloud-core==2.4.3 +google-cloud-logging==3.12.1 +google-cloud-monitoring==2.27.2 +google-cloud-resource-manager==1.14.2 +google-cloud-storage==2.19.0 +google-crc32c==1.7.1 +google-genai==1.27.0 +google-pasta==0.2.0 +google-resumable-media==2.7.2 +googleapis-common-protos==1.70.0 +grain==0.2.11 +graphviz==0.21 +grpc-google-iam-v1==0.14.2 +grpcio==1.74.0 +grpcio-status==1.74.0 +gviz-api==1.10.0 +h11==0.16.0 +h5py==3.14.0 +hf-xet==1.1.5 +hf_transfer==0.1.9 +httpcore==1.0.9 +httpx==0.28.1 +huggingface-hub==0.34.1 +humanize==4.12.3 +idna==3.10 imageio==2.37.0 imageio-ffmpeg==0.6.0 -hf_transfer>=0.1.9 \ No newline at end of file +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.2.1 +jax==0.7.0 +jaxlib==0.7.0 +Jinja2==3.1.6 +keras==3.10.0 +lazy_loader==0.4 +libclang==18.1.1 +libcst==1.8.2 +libtpu==0.0.19 +Markdown==3.8.2 +markdown-it-py==3.0.0 +MarkupSafe==3.0.2 +mccabe==0.7.0 +mdurl==0.1.2 +ml_collections==1.1.0 +ml_dtypes==0.5.1 +ml_goodput_measurement==0.0.13 +more-itertools==10.7.0 +mpmath==1.3.0 +msgpack==1.1.1 +multidict==6.6.3 +multiprocess==0.70.16 +mypy_extensions==1.1.0 +namex==0.1.0 +nest-asyncio==1.6.0 +networkx==3.5 +numpy==2.1.3 +oauthlib==3.3.1 +opencv-python-headless==4.10.0.84 +opentelemetry-api==1.35.0 +opt_einsum==3.4.0 +optax==0.2.5 +optree==0.17.0 +orbax-checkpoint==0.11.19 +packaging==25.0 +pandas==2.3.1 +parameterized==0.9.0 +pathspec==0.12.1 +pillow==11.3.0 +platformdirs==4.3.8 +pluggy==1.6.0 +promise==2.3 +propcache==0.3.2 +proto-plus==1.26.1 +protobuf==5.29.5 +psutil==7.0.0 +pyarrow==21.0.0 +pyasn1==0.6.1 +pyasn1_modules==0.4.2 +pydantic==2.11.7 +pydantic_core==2.33.2 +Pygments==2.19.2 +pyink==24.10.1 +pylint==3.3.7 +pytest==8.2.2 +python-dateutil==2.9.0.post0 +pytz==2025.2 +PyYAML==6.0.2 +regex==2024.11.6 +requests==2.32.4 +requests-oauthlib==2.0.0 +rich==14.1.0 +rsa==4.9.1 +ruff==0.2.0 +safetensors==0.5.3 +scikit-image==0.25.2 +scipy==1.16.0 +sentencepiece==0.2.0 +setuptools==80.9.0 +shapely==2.1.1 +simple-parsing==0.1.7 +simplejson==3.20.1 +six==1.17.0 +sniffio==1.3.1 +sympy==1.14.0 +tenacity==8.5.0 +tensorboard==2.19.0 +tensorboard-data-server==0.7.2 +tensorboard_plugin_profile==2.20.4 +tensorboardX==2.6.4 +tensorflow==2.19.0 +tensorflow-datasets==4.9.9 +tensorflow-metadata==1.17.2 +tensorstore==0.1.76 +termcolor==3.1.0 +tifffile==2025.6.11 +tokenizers==0.21.0 +toml==0.10.2 +tomlkit==0.13.3 +toolz==1.0.0 +torch==2.7.1+cpu +torchvision==0.22.1+cpu +tqdm==4.67.1 +transformers==4.48.1 +treescope==0.1.9 +typing-inspection==0.4.1 +typing_extensions==4.14.1 +tzdata==2025.2 +urllib3==2.5.0 +wcwidth==0.2.13 +websockets==15.0.1 +Werkzeug==3.1.3 +wheel==0.45.1 +wrapt==1.17.2 +xprof==2.20.4 +xxhash==3.5.0 +yarl==1.20.1 +zipp==3.23.0 \ No newline at end of file