Update requirements.txt

requirements.txt

@@ -1,239 +1,3 @@
-absl-py==2.1.0
-accelerate==0.27.2
-aiofiles==23.2.1
-aiohttp==3.9.3
-aiosignal==1.3.1
-alembic==1.13.1
-altair==5.3.0
-annotated-types==0.6.0
-anyio==4.3.0
-appdirs==1.4.4
-APScheduler==3.10.4
-archspec @ file:///croot/archspec_1709217642129/work
-argon2-cffi==23.1.0
-argon2-cffi-bindings==21.2.0
-arrow==1.3.0
-asttokens==2.4.1
-async-lru==2.0.4
-attrs==23.2.0
-autotransformers==0.0.7
-Babel==2.14.0
-beautifulsoup4==4.12.3
-bitsandbytes==0.42.0
-bleach==6.1.0
-blinker==1.7.0
-boltons @ file:///work/ci_py311/boltons_1677685195580/work
-cachetools==5.3.3
-certifi @ file:///croot/certifi_1707229174982/work/certifi
-cffi @ file:///croot/cffi_1700254295673/work
-charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work
-click==8.1.7
-colorlog==6.8.2
-comm==0.2.2
-conda @ file:///croot/conda_1710772050586/work
-conda-libmamba-solver @ file:///croot/conda-libmamba-solver_1706733287605/work/src
-conda-package-handling @ file:///croot/conda-package-handling_1690999929514/work
-conda_package_streaming @ file:///croot/conda-package-streaming_1690987966409/work
-contourpy==1.2.1
-cycler==0.12.1
-datasets==2.18.0
-debugpy==1.8.1
-decorator==5.1.1
-defusedxml==0.7.1
-dill==0.3.8
-distro @ file:///croot/distro_1701455004953/work
-docker-pycreds==0.4.0
-docstring_parser==0.16
-einops==0.6.1
-evaluate==0.2.2
-executing==2.0.1
-fastapi==0.110.1
-fastjsonschema==2.19.1
-ffmpy==0.3.2
-filelock==3.13.3
-fonttools==4.51.0
-fqdn==1.5.1
-frozenlist==1.4.1
-fsspec==2024.2.0
-gdown==5.1.0
-gitdb==4.0.11
-GitPython==3.1.43
-gradio==4.25.0
-gradio_client==0.15.0
-greenlet==3.0.3
-grpcio==1.62.1
-h11==0.14.0
-httpcore==1.0.5
-httpx==0.27.0
-huggingface-hub==0.22.2
-idna @ file:///work/ci_py311/idna_1676822698822/work
-importlib_resources==6.4.0
-iniconfig==2.0.0
-ipykernel==6.29.4
-ipython==8.23.0
-ipywidgets==8.1.2
-isoduration==20.11.0
-jedi==0.19.1
-Jinja2==3.1.3
-joblib==1.4.0
-json5==0.9.24
-jsonpatch @ file:///croot/jsonpatch_1710807507480/work
-jsonpointer==2.1
-jsonschema==4.21.1
-jsonschema-specifications==2023.12.1
-jupyter-events==0.10.0
-jupyter-lsp==2.2.4
-jupyter_client==8.6.1
-jupyter_core==5.7.2
-jupyter_server==2.13.0
-jupyter_server_terminals==0.5.3
-jupyterlab==4.1.6
-jupyterlab_pygments==0.3.0
-jupyterlab_server==2.26.0
-jupyterlab_widgets==3.0.10
-kiwisolver==1.4.5
-libmambapy @ file:///croot/mamba-split_1712091911343/work/libmambapy
-Mako==1.3.2
-Markdown==3.6
-markdown-it-py==3.0.0
-MarkupSafe==2.1.5
-matplotlib==3.8.4
-matplotlib-inline==0.1.6
-mdurl==0.1.2
-menuinst @ file:///croot/menuinst_1706732933928/work
-mistune==3.0.2
-more-itertools==8.14.0
-mpmath==1.3.0
-multidict==6.0.5
-multiprocess==0.70.16
-nbclient==0.10.0
-nbconvert==7.16.3
-nbformat==5.10.4
-nest-asyncio==1.6.0
-networkx==3.3
-nlpaug==1.1.11
-nltk==3.8.1
-notebook_shim==0.2.4
-numpy==1.26.4
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu12==11.0.2.54
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.19.3
-nvidia-nvjitlink-cu12==12.4.127
-nvidia-nvtx-cu12==12.1.105
-optuna==3.6.1
-orjson==3.10.0
-overrides==7.7.0
-packaging @ file:///croot/packaging_1710807400464/work
-pandas==1.5.3
-pandocfilters==1.5.1
-parso==0.8.4
-peft==0.10.0
-pexpect==4.9.0
-pillow==10.3.0
-platformdirs @ file:///croot/platformdirs_1692205439124/work
-pluggy @ file:///work/ci_py311/pluggy_1676822818071/work
-polyfuzz==0.4.2
-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
-pyarrow-hotfix==0.6
-pycosat @ file:///croot/pycosat_1696536503704/work
-pycparser @ file:///tmp/build/80754af9/pycparser_1636541352034/work
-pydantic==2.6.4
-pydantic_core==2.16.3
-pydeck==0.8.1b0
-pydub==0.25.1
-Pygments==2.17.2
-pyparsing==3.1.2
-PySocks==1.7.1
-pytest==7.4.4
-python-dateutil==2.9.0.post0
-python-json-logger==2.0.7
-python-multipart==0.0.9
-pytz==2024.1
-PyYAML==6.0.1
-pyzmq==25.1.2
-rapidfuzz==3.8.1
-referencing==0.34.0
-regex==2023.12.25
-requests @ file:///croot/requests_1707355572290/work
-responses==0.18.0
-rfc3339-validator==0.1.4
-rfc3986-validator==0.1.1
-rich==13.7.1
-rouge-score==0.1.2
-rpds-py==0.18.0
-ruamel.yaml @ file:///work/ci_py311/ruamel.yaml_1676838772170/work
-ruff==0.3.5
-safetensors==0.4.2
-scikit-learn==1.4.1.post1
-scipy==1.13.0
-seaborn==0.12.2
-semantic-version==2.10.0
-Send2Trash==1.8.3
-sentencepiece==0.1.99
-sentry-sdk==1.44.1
-seqeval==1.2.2
-setproctitle==1.3.3
-shellingham==1.5.4
-shtab==1.7.1
-simpletransformers==0.61.14
-six==1.16.0
-smmap==5.0.1
-sniffio==1.3.1
-soupsieve==2.5
-SQLAlchemy==2.0.29
-stack-data==0.6.3
-starlette==0.37.2
-streamlit==1.33.0
-sympy==1.12
-tenacity==8.2.3
-tensorboard==2.16.2
-tensorboard-data-server==0.7.2
-tensorboardX==2.6.2.2
-terminado==0.18.1
-threadpoolctl==3.4.0
-tinycss2==1.2.1
-tokenizers==0.15.2
-toml==0.10.2
-tomlkit==0.12.0
-toolz==0.12.1
-torch==2.2.2
-tornado==6.4
-tqdm @ file:///croot/tqdm_1679561862951/work
-traitlets==5.14.2
 transformers==4.38.2
-triton==2.2.0
-trl==0.7.11
-truststore @ file:///croot/truststore_1695244293384/work
-typer==0.12.1
-types-python-dateutil==2.9.0.20240316
-typing_extensions==4.11.0
-tyro==0.8.2
-tzlocal==5.2
-uri-template==1.3.0
-urllib3 @ file:///croot/urllib3_1700840488771/work
-uvicorn==0.29.0
-wandb==0.16.6
-watchdog==4.0.0
-wcwidth==0.2.13
-webcolors==1.13
-webencodings==0.5.1
-websocket-client==1.7.0
-websockets==11.0.3
-Werkzeug==3.0.2
-widgetsnbextension==4.0.10
-xxhash==3.4.1
-yarl==1.9.4
-zstandard @ file:///work/ci_py311_2/zstandard_1679339489613/work
+torch==2.2.2+cu121
+gradio==4.25.0