Update requirements.txt

requirements.txt

@@ -1,321 +1,27 @@
-absl-py==2.1.0
+pytest
+matplotlib
+scikit-learn
+scipy
+spacy
+numpy
+einops
+einsum
+fvcore
+h5py
+twine
+transformers==4.45.2
+huggingface_hub==0.24
 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
-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
-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
+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].3#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