Update requirements.txt

requirements.txt

@@ -1,287 +1,14 @@
-absl-py==2.1.0
-accelerate==0.29.3
-aiofiles==23.2.1
-aiohttp==3.9.3
-aiosignal==1.3.1
-altair==5.3.0
-annotated-types==0.6.0
-antlr4-python3-runtime==4.9.3
-anyio==4.3.0
-argon2-cffi==23.1.0
-arrow==1.3.0
-asttokens==2.4.1
-async-lru==2.0.4
-async-timeout==4.0.3
-attrs==23.2.0
-audiocraft==1.2.0
-audioread==3.0.1
-av==12.0.0
-Babel==2.14.0
-backcall==0.2.0
-beautifulsoup4==4.12.3
-bleach==6.1.0
-blessed==1.20.0
-blis==0.7.11
-cachetools==5.3.3
-calmsize==0.1.3
-catalogue==2.0.10
-certifi==2024.2.2
-cffi==1.16.0
-charset-normalizer==3.3.2
-click==8.1.7
-cloudpathlib==0.16.0
-cloudpickle==3.0.0
-cmake==3.29.2
-colorama==0.4.6
-colorlog==6.8.2
-comm==0.2.1
-confection==0.1.4
-contourpy==1.1.1
-cycler==0.12.1
-cymem==2.0.8
-debugpy==1.8.1
-decorator==5.1.1
-demucs==4.0.1
-docker-pycreds==0.4.0
-docopt==0.6.2
-dora_search==0.1.12
-einops==0.7.0
-encodec==0.1.1
-exceptiongroup==1.2.0
-executing==2.0.1
-fastapi==0.110.1
-fastjsonschema==2.19.1
-ffmpy==0.3.2
-filelock==3.13.4
-flash-attn==2.5.9.post1
-flashy==0.0.2
-fonttools==4.49.0
-fqdn==1.5.1
-frozenlist==1.4.1
-fsspec==2024.3.1
-ftfy==6.2.0
-future==1.0.0
-gitdb==4.0.11
-GitPython==3.1.43
-google-auth==2.28.2
-google-auth-oauthlib==1.0.0
-gpustat==1.1.1
-grpcio==1.62.1
-h11==0.14.0
-httpcore==1.0.4
-httpx==0.27.0
-huggingface-hub==0.20.2
-hydra-colorlog==1.2.0
-hydra-core==1.3.2
-icecream==2.1.3
-idna==3.6
-imageio==2.34.1
-importlib_metadata==7.0.2
-importlib_resources==6.1.3
-ipdb==0.13.13
-ipykernel==6.29.3
-ipython==8.12.0
-ipywidgets==8.1.2
-isoduration==20.11.0
-jedi==0.19.1
-Jinja2==3.1.3
-joblib==1.4.0
-json5==0.9.22
-jsonpointer==2.4
-jsonschema==4.21.1
-jsonschema-specifications==2023.12.1
-julius==0.2.7
+torchlibrosa==0.1.0
+pytorch-lightning==1.7.0
 jupyter==1.0.0
-jupyter-console==6.6.3
-jupyter-events==0.9.0
-jupyter-lsp==2.2.4
-jupyter_client==8.6.0
-jupyter_core==5.7.1
-jupyter_server==2.13.0
-jupyter_server_terminals==0.5.2
-jupyterlab==4.1.4
-jupyterlab_pygments==0.3.0
-jupyterlab_server==2.25.3
-jupyterlab_widgets==3.0.10
-kiwisolver==1.4.5
-lameenc==1.7.0
-langcodes==3.3.0
-lazy_loader==0.3
-librosa==0.9.1
-lightning-utilities==0.11.2
-lit==18.1.3
-llvmlite==0.41.1
-Markdown==3.5.2
-markdown-it-py==3.0.0
-MarkupSafe==2.1.5
-matplotlib==3.7.5
-matplotlib-inline==0.1.6
-mdurl==0.1.2
-mistune==3.0.2
-mpmath==1.3.0
-msgpack==1.0.8
-multidict==6.0.5
-murmurhash==1.0.10
-mypy-extensions==1.0.0
-nbclient==0.9.0
-nbconvert==7.16.2
-nbformat==5.9.2
-nest-asyncio==1.6.0
-networkx==3.1
-notebook==7.1.1
-notebook_shim==0.2.4
-num2words==0.5.13
-numba==0.58.1
-numpy==1.24.4
-nvidia-cublas-cu11==11.10.3.66
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu11==11.7.101
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu11==11.7.99
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu11==11.7.99
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu11==8.5.0.96
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu11==10.9.0.58
-nvidia-cufft-cu12==11.0.2.54
-nvidia-curand-cu11==10.2.10.91
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu11==11.4.0.1
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu11==11.7.4.91
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-ml-py==12.535.133
-nvidia-nccl-cu11==2.14.3
-nvidia-nccl-cu12==2.20.5
-nvidia-nvjitlink-cu12==12.4.127
-nvidia-nvtx-cu11==11.7.91
-nvidia-nvtx-cu12==12.1.105
-oauthlib==3.2.2
-omegaconf==2.3.0
-opencv-python==4.9.0.80
-openunmix==1.2.1
-orjson==3.10.0
-overrides==7.7.0
-packaging==23.2
-pandas==2.0.3
-pandocfilters==1.5.1
-parso==0.8.4
-pexpect==4.9.0
-pickleshare==0.7.5
-Pillow==9.5.0
-pkgutil_resolve_name==1.3.10
-platformdirs==4.2.0
-pooch==1.8.1
-preshed==3.0.9
-prometheus_client==0.20.0
-prompt-toolkit==3.0.43
-protobuf==4.25.3
-psutil==5.9.8
-ptyprocess==0.7.0
-pure-eval==0.2.2
-pyarrow==15.0.2
-pyasn1==0.5.1
-pyasn1-modules==0.3.0
-pycparser==2.21
-pydantic==2.6.4
-pydantic_core==2.16.3
-pyDeprecate==0.3.1
-pydub==0.25.1
-Pygments==2.17.2
-pyparsing==3.1.2
-PyPDF2==2.12.1
-pyre-extensions==0.0.23
-python-dateutil==2.9.0.post0
-python-json-logger==2.0.7
-python-multipart==0.0.9
-pytorch-lightning==1.9.0
+icecream
+torchmetrics==0.11.4
+huggingface_hub==0.20.2
+calmsize
+librosa==0.10.1
+einops==0.7.0
 pytorch-memlab==0.3.0
-pytz==2024.1
-PyWavelets==1.4.1
-PyYAML==6.0.1
-pyzmq==25.1.2
-qtconsole==5.5.1
-QtPy==2.4.1
-referencing==0.33.0
-regex==2024.4.16
-requests==2.31.0
-requests-oauthlib==2.0.0
-resampy==0.4.3
-retrying==1.3.4
-rfc3339-validator==0.1.4
-rfc3986-validator==0.1.1
-rich==13.7.1
-rpds-py==0.18.0
-rsa==4.9
-ruff==0.3.5
-safetensors==0.4.3
-scikit-image==0.21.0
-scikit-learn==1.3.2
-scipy==1.10.1
-semantic-version==2.10.0
-Send2Trash==1.8.2
-sentencepiece==0.2.0
-sentry-sdk==2.1.1
-setproctitle==1.3.3
-shellingham==1.5.4
-six==1.16.0
-smart-open==6.4.0
-smmap==5.0.1
-sniffio==1.3.1
-SoundFile==0.10.3.post1
-soupsieve==2.5
-soxr==0.3.7
-spacy==3.7.4
-spacy-legacy==3.0.12
-spacy-loggers==1.0.5
-srsly==2.4.8
-ssr-eval==0.0.7
-stack-data==0.6.3
-starlette==0.37.2
-submitit==1.5.1
-sympy==1.12
-taming-transformers-rom1504==0.0.6
-tensorboard==2.14.0
-tensorboard-data-server==0.7.2
-terminado==0.18.0
-thinc==8.2.3
-threadpoolctl==3.4.0
-tifffile==2023.7.10
-tinycss2==1.2.1
-tokenizers==0.15.2
-tomli==2.0.1
-tomlkit==0.12.0
-toolz==0.12.1
-torch==2.0.1+cu117
-torchaudio==0.12.1+cu113
-torchlibrosa==0.1.0
-torchmetrics==1.3.2
-torchvision==0.13.1+cu113
-tornado==6.4
-tqdm==4.66.2
-traitlets==5.14.1
 transformers==4.39.0
-treetable==0.2.5
-triton==2.0.0
-typer==0.9.4
-types-python-dateutil==2.8.19.20240106
-typing-inspect==0.9.0
-typing_extensions==4.11.0
-tzdata==2024.1
-uri-template==1.3.0
-urllib3==2.2.1
-uvicorn==0.29.0
-wandb==0.17.0
-wasabi==1.1.2
-Wave==0.0.2
-wcwidth==0.2.13
-weasel==0.3.4
-webcolors==1.13
-websocket-client==1.7.0
-websockets==11.0.3
-Werkzeug==3.0.1
-widgetsnbextension==4.0.10
-xformers==0.0.22
-yarl==1.9.4
-zipp==3.17.0
+omegaconf==2.3.0
+sentencepiece==0.2.0
+taming-transformers-rom1504