Update requirements.txt

requirements.txt

@@ -1,54 +1,321 @@
-torch
-torchvision
-pytest
-matplotlib
-jupyter
-ipython
-ipdb
-gpustat
-scikit-learn
-spacy
-munch
-einops
-opt_einsum
-fvcore
-gsutil
-cmake
-pykeops
-zstandard
-psutil
-h5py
-twine
-gdown
-hydra-core==1.3.1
-hydra-colorlog==1.2.0
-hydra-optuna-sweeper==1.2.0
-pyrootutils
-rich
-transformers==4.45.2
+absl-py==2.1.0
+accelerate==0.34.2
+addict==2.4.0
+aiofiles==23.2.1
+aiohttp==3.9.5
+aiosignal==1.3.1
+albumentations==1.4.0
+altair==5.3.0
+annotated-types==0.6.0
+antlr4-python3-runtime==4.9.3
+anyio==4.3.0
+appdirs==1.4.4
+array_record==0.5.1
+asttokens==2.4.1
+astunparse==1.6.3
+async-timeout==4.0.3
+attrs==23.2.0
+av==11.0.0
+backcall==0.2.0
+bcrypt==4.2.0
+beautifulsoup4==4.12.3
+braceexpand==0.1.7
+Brotli==1.0.9
+cachetools==5.3.3
+certifi==2024.2.2
+cffi==1.17.1
+chardet==5.2.0
+charset-normalizer==3.3.2
+chex==0.1.86
+click==8.1.7
+clip-benchmark==1.6.1
+cloudpickle==3.0.0
+clu==0.0.12
+colorlog==6.8.2
+comm==0.2.2
+contextlib2==21.6.0
+contourpy==1.2.1
+controlnet-aux==0.0.9
+cryptography==43.0.1
+cycler==0.12.1
+dataclasses==0.6
 datasets==3.0.1
+debugpy==1.6.7
+decorator==4.4.2
+decord==0.6.0
+deepspeed==0.15.2
+defusedxml==0.7.1
+diffrax==0.6.0
+diffusers==0.30.3
+dill==0.3.8
+dm-tree==0.1.8
+docker-pycreds==0.4.0
+docstring_parser==0.16
+editdistance==0.8.1
+einops==0.7.0
+equinox==0.11.7
+etils==1.5.2
+exceptiongroup==1.2.1
+executing==2.1.0
+ExifRead-nocycle==3.0.1
+fabric==3.0.1
+fairscale==0.4.13
+fastapi==0.110.0
+ffmpeg-python==0.2.0
+ffmpy==0.3.2
+filelock==3.13.1
+fire==0.5.0
+flash-attn==2.6.3
+flatbuffers==24.3.25
+flax==0.8.5
+flaxmodels==0.1.3
+fonttools==4.51.0
+frasch==0.1
+frozenlist==1.4.1
+fsspec==2024.3.1
+ftfy==6.1.3
+future==1.0.0
+fvcore==0.1.5.post20221221
+gast==0.4.0
+gcsfs==2024.3.1
+gdown==5.1.0
+gitdb==4.0.11
+GitPython==3.1.43
+gmpy2==2.1.2
+google-api-core==2.18.0
+google-auth==2.29.0
+google-auth-oauthlib==1.0.0
+google-cloud-core==2.4.1
+google-cloud-storage==2.16.0
+google-crc32c==1.5.0
+google-pasta==0.2.0
+google-resumable-media==2.7.0
+googleapis-common-protos==1.63.0
+gradio==4.44.1
+gradio_client==1.3.0
+grpcio==1.62.2
+h11==0.14.0
+h5py==3.10.0
+hjson==3.1.0
+httpcore==1.0.5
+httpx==0.27.0
+huggingface-hub==0.26.3
+idna==3.6
+imageio==2.34.0
+imageio-ffmpeg==0.4.9
+img2dataset==1.45.0
+importlib_metadata==7.1.0
+importlib_resources==6.4.0
+invoke==2.2.0
+iopath==0.1.10
+ipdb==0.13.13
+ipykernel==6.29.5
+ipython==8.12.0
+jax==0.4.30
+jax-cuda12-pjrt==0.4.30
+jax-cuda12-plugin==0.4.30
+jaxlib==0.4.30
+jaxtyping==0.2.34
+jedi==0.19.2
+Jinja2==3.1.3
+joblib==1.4.0
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+keras==2.12.0
+kiwisolver==1.4.5
+lazy_loader==0.4
+libclang==18.1.1
+lightning-utilities==0.11.2
+lineax==0.0.5
+lpips==0.1.4
+Markdown==3.6
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.7.5
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+mediapipe==0.10.18
+mkl_fft==1.3.10
+mkl_random==1.2.7
+mkl-service==2.4.0
+ml-collections==0.1.1
+ml-dtypes==0.4.0
+mlxu==0.2.0
+mmcv-full==1.7.2
+mmdet==2.28.2
+moviepy==1.0.3
+mpmath==1.3.0
+msgpack==1.0.8
+multidict==6.0.5
+multiprocess==0.70.16
+nest_asyncio==1.6.0
+networkx==3.2.1
+ninja==1.11.1.1
+numpy==1.23.2
+nvidia-cublas-cu11==11.11.3.6
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu11==11.8.87
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvcc-cu12==12.6.68
+nvidia-cuda-nvrtc-cu11==11.8.89
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu11==11.8.89
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu11==8.7.0.84
+nvidia-cudnn-cu12==9.4.0.58
+nvidia-cufft-cu11==10.9.0.58
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu11==10.3.0.86
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu11==11.4.1.48
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu11==11.7.5.86
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-ml-py==12.535.161
+nvidia-nccl-cu11==2.20.5
+nvidia-nccl-cu12==2.18.1
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu11==11.8.86
+nvidia-nvtx-cu12==12.1.105
+nvitop==1.3.2
+oauthlib==3.2.2
+omegaconf==2.3.0
+open_clip_torch==2.29.0
+opencv-contrib-python==4.10.0.84
+opencv-python==4.9.0.80
+opencv-python-headless==4.9.0.80
+opt-einsum==3.3.0
+optax==0.2.2
+optimistix==0.0.7
+orbax-checkpoint==0.5.20
+orjson==3.10.1
+packaging==24.0
+pandas==2.0.3
+parameterized==0.9.0
+paramiko==3.5.0
+parso==0.8.4
+peft==0.13.2
+pexpect==4.9.0
+pickleshare==0.7.5
+pillow==10.2.0
+pip==23.3.1
+platformdirs==4.3.6
+portalocker==2.8.2
+proglog==0.1.10
+promise==2.3
+prompt_toolkit==3.0.48
+proto-plus==1.23.0
+protobuf==4.25.5
+psutil==5.9.8
+ptyprocess==0.7.0
+pure_eval==0.2.3
+py-cpuinfo==9.0.0
+pyarrow==15.0.2
+pyasn1==0.6.0
+pyasn1_modules==0.4.0
+pycocoevalcap==1.2
+pycocotools==2.0.8
+pycparser==2.22
+pydantic==2.6.4
+pydantic_core==2.16.3
+pydub==0.25.1
+pyglove==0.4.4
+Pygments==2.17.2
+PyNaCl==1.5.0
+pyOpenSSL==24.2.1
+pyparsing==3.1.2
+PySocks==1.7.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.9
 pytorch-lightning==2.2.1
-triton==2.3.1
-wandb==0.16.3
-controlnet_aux==0.0.9
+pytorch3d==0.7.8
+pytz==2024.1
+PyYAML==6.0.1
+pyzmq==26.2.0
+qudida==0.0.4
+referencing==0.34.0
+regex==2023.12.25
+requests==2.32.3
+requests-oauthlib==2.0.0
+rich==13.7.1
+rpds-py==0.18.0
+rsa==4.9
+ruff==0.7.2
+safetensors==0.4.3
+scikit-image==0.22.0
+scikit-learn==1.3.2
+scikit-video==1.1.11
+scipy==1.10.1
+semantic-version==2.10.0
+sentencepiece==0.1.99
+sentry-sdk==1.45.0
+seqio==0.0.19
+setproctitle==1.3.3
+setuptools==68.2.2
+shellingham==1.5.4
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.1
+sounddevice==0.5.1
+soupsieve==2.5
+stack-data==0.6.2
+starlette==0.36.3
+supervision==0.25.0
+sympy==1.12
+tabulate==0.9.0
+tensorboard==2.12.3
+tensorboard-data-server==0.7.2
+tensorflow==2.12.1
+tensorflow-estimator==2.12.0
+tensorflow-hub==0.16.1
+tensorflow-io==0.32.0
+tensorflow-io-gcs-filesystem==0.32.0
+tensorflow-metadata==1.15.0
+tensorflow-text==2.12.0
+tensorstore==0.1.62
+termcolor==2.4.0
+terminaltables==3.1.10
+test-tube==0.7.5
+tf-keras==2.15.0
+tfds-nightly==4.9.2.dev202308090034
+threadpoolctl==3.4.0
+tifffile==2024.4.18
 timm==0.6.7
+tokenizers==0.20.0
+toml==0.10.2
+tomli==2.1.0
+tomlkit==0.12.0
+toolz==0.12.1
+torch==2.3.1+cu118
+torchaudio==2.3.1+cu118
+torchdiffeq==0.2.3
 torchmetrics==1.3.2
-git+https://github.com/mlcommons/logging.git@2.1.0
-accelerate==0.34.2
-diffusers==0.30.3
-deepspeed==0.15.2
-sentencepiece==0.1.99
-pillow==10.2.0
-torch==2.3.1
-torchvision==0.18.1
-torchaudio==2.3.1
-flash-attn==2.6.3
-git+https://github.com/Dao-AILab/[email protected]#subdirectory=csrc/fused_dense_lib
-jaxtyping
-mediapipe
-gradio
-git+https://github.com/facebookresearch/pytorch3d.git
-opencv-python==4.5.5.64
-opencv-python-headless==4.5.5.64
-huggingface_hub==0.24
-numpy==1.24.4
+torchvision==0.18.1+cu118
+tordi==1.0.0
+tornado==6.2
+tqdm==4.66.5
+traitlets==5.14.3
+transformers==4.47.0.dev0
+triton==2.3.1
+typeguard==2.13.3
+typer==0.12.3
+typing_extensions==4.12.2
+tzdata==2024.1
+urllib3==2.2.1
+uvicorn==0.27.1
+wandb==0.16.3
+wcwidth==0.2.13
+webcolors==24.11.1
+webdataset==0.2.86
+websockets==11.0.3
+Werkzeug==3.0.2
+wheel==0.41.2
+wrapt==1.14.1
+xformers==0.0.22.post7
+xxhash==3.5.0
+yacs==0.1.8
+yapf==0.40.2
+yarl==1.9.4
+zipp==3.18.1