updated requirements.txt

requirements.txt

@@ -1,161 +1,5 @@
-accelerate==0.34.2
-aiofiles==23.2.1
-aiohappyeyeballs==2.4.0
-aiohttp==3.10.6
-aiosignal==1.3.1
-annotated-types==0.7.0
-anyio==4.6.0
-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==24.2.0
-babel==2.16.0
-beautifulsoup4==4.12.3
-bleach==6.1.0
-certifi==2024.8.30
-cffi==1.17.1
-charset-normalizer==3.3.2
-click==8.1.7
-comm==0.2.2
-contourpy==1.3.0
-cycler==0.12.1
-datasets==3.0.1
-debugpy==1.8.6
-decorator==5.1.1
-defusedxml==0.7.1
-dill==0.3.8
-evaluate==0.4.3
-executing==2.1.0
-fastapi==0.115.0
-fastjsonschema==2.20.0
-ffmpy==0.4.0
-filelock==3.13.1
-fonttools==4.54.1
-fqdn==1.5.1
-frozenlist==1.4.1
-fsspec==2024.2.0
-gradio==4.44.0
-gradio_client==1.3.0
-h11==0.14.0
-httpcore==1.0.5
-httpx==0.27.2
-huggingface-hub==0.25.1
-idna==3.10
-importlib_resources==6.4.5
-ipykernel==6.29.5
-ipython==8.27.0
-isoduration==20.11.0
-jedi==0.19.1
-Jinja2==3.1.3
-joblib==1.4.2
-json5==0.9.25
-jsonpointer==3.0.0
-jsonschema==4.23.0
-jsonschema-specifications==2023.12.1
-jupyter-events==0.10.0
-jupyter-lsp==2.2.5
-jupyter_client==8.6.3
-jupyter_core==5.7.2
-jupyter_server==2.14.2
-jupyter_server_terminals==0.5.3
-jupyterlab==4.2.5
-jupyterlab_pygments==0.3.0
-jupyterlab_server==2.27.3
-kiwisolver==1.4.7
-lightning==2.4.0
-lightning-utilities==0.11.7
-markdown-it-py==3.0.0
-MarkupSafe==2.1.5
-matplotlib==3.9.2
-matplotlib-inline==0.1.7
-mdurl==0.1.2
-mistune==3.0.2
-mpmath==1.3.0
-multidict==6.1.0
-multiprocess==0.70.16
-nbclient==0.10.0
-nbconvert==7.16.4
-nbformat==5.10.4
-nest-asyncio==1.6.0
-networkx==3.2.1
-notebook_shim==0.2.4
-numpy==1.26.3
-orjson==3.10.7
-overrides==7.7.0
-packaging==24.1
-pandas==2.2.3
-pandocfilters==1.5.1
-parso==0.8.4
-pexpect==4.9.0
-pillow==10.2.0
-platformdirs==4.3.6
-prometheus_client==0.21.0
-prompt_toolkit==3.0.48
-psutil==6.0.0
-ptyprocess==0.7.0
-pure_eval==0.2.3
-pyarrow==17.0.0
-pycparser==2.22
-pydantic==2.9.2
-pydantic_core==2.23.4
-pydub==0.25.1
-Pygments==2.18.0
-pyparsing==3.1.4
-python-dateutil==2.9.0.post0
-python-json-logger==2.0.7
-python-multipart==0.0.10
-pytorch-lightning==2.4.0
-pytz==2024.2
-PyYAML==6.0.2
-pyzmq==26.2.0
-referencing==0.35.1
-regex==2024.9.11
-requests==2.32.3
-rfc3339-validator==0.1.4
-rfc3986-validator==0.1.1
-rich==13.8.1
-rpds-py==0.20.0
-ruff==0.6.7
-safetensors==0.4.5
-scikit-learn==1.5.2
-scipy==1.14.1
-seaborn==0.13.2
-semantic-version==2.10.0
-Send2Trash==1.8.3
-sentencepiece==0.2.0
-shellingham==1.5.4
-six==1.16.0
-sniffio==1.3.1
-soupsieve==2.6
-stack-data==0.6.3
-starlette==0.38.6
-sympy==1.12
-terminado==0.18.1
-threadpoolctl==3.5.0
-tinycss2==1.3.0
-tokenizers==0.20.0
-tomlkit==0.12.0
-torch==2.4.1+cpu
-torchaudio==2.4.1+cpu
-torchmetrics==1.4.2
-torchvision==0.19.1+cpu
-tornado==6.4.1
-tqdm==4.66.5
-traitlets==5.14.3
-transformers==4.45.0
-typer==0.12.5
-types-python-dateutil==2.9.0.20240906
-typing_extensions==4.9.0
-tzdata==2024.2
-uri-template==1.3.0
-urllib3==2.2.3
-uvicorn==0.30.6
-wcwidth==0.2.13
-webcolors==24.8.0
-webencodings==0.5.1
-websocket-client==1.8.0
-websockets==12.0
-xxhash==3.5.0
-yarl==1.12.1
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
+lightning 
+transformers 
+accelerate