summaryrefslogtreecommitdiff
path: root/requirements.txt
diff options
context:
space:
mode:
Diffstat (limited to 'requirements.txt')
-rw-r--r--requirements.txt283
1 files changed, 20 insertions, 263 deletions
diff --git a/requirements.txt b/requirements.txt
index fca3779..06637df 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,274 +1,31 @@
-absl-py==2.1.0
accelerate==0.33.0
-aiofiles==23.2.1
-annotated-types==0.6.0
-anyio==4.4.0
-argon2-cffi==23.1.0
-argon2-cffi-bindings==21.2.0
-arrow==1.3.0
-asttokens==2.4.1
-astunparse==1.6.3
-async-lru==2.0.4
-attrs==23.2.0
-audioread==3.0.1
-babel==2.16.0
-beautifulsoup4==4.12.3
-bitsandbytes==0.43.3
-bleach==6.1.0
-blis==0.7.11
-cachetools==5.3.2
-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
-cmake==3.28.1
-comm==0.2.1
-confection==0.1.4
-contourpy==1.2.0
-cycler==0.12.1
-cymem==2.0.8
-Cython==3.0.8
datasets==2.21.0
-debugpy==1.8.1
-decorator==5.1.1
-defusedxml==0.7.1
-dill==0.3.8
-diskcache==5.6.3
-distro==1.9.0
-dm-tree==0.1.8
-docstring_parser==0.16
-einops==0.7.0
-exceptiongroup==1.2.0
-execnet==2.0.2
-executing==2.0.1
-expecttest==0.1.3
-fastapi==0.112.2
-fastjsonschema==2.19.1
-ffmpy==0.4.0
-filelock==3.13.1
-fire==0.6.0
-fonttools==4.48.1
-fqdn==1.5.1
-gast==0.5.4
-gguf==0.9.1
-google-auth==2.27.0
-google-auth-oauthlib==0.4.6
-gradio==4.42.0
-gradio_client==1.3.0
-grpcio==1.60.1
-h11==0.14.0
-hjson==3.1.0
-httpcore==1.0.5
-httptools==0.6.1
-httpx==0.27.2
-huggingface-hub==0.24.6
-hypothesis==5.35.1
-idna==3.6
-importlib_resources==6.4.4
-iniconfig==2.0.0
-intel-openmp==2021.4.0
-interegular==0.3.3
-ipykernel==6.29.2
-ipython==8.21.0
-ipython-genutils==0.2.0
-isoduration==20.11.0
-jedi==0.19.1
-jieba==0.42.1
-Jinja2==3.1.3
-jiter==0.5.0
-joblib==1.3.2
-json5==0.9.14
-jsonpointer==3.0.0
-jsonschema==4.21.1
-jsonschema-specifications==2023.12.1
-jupyter-events==0.10.0
-jupyter-lsp==2.2.5
-jupyter_client==8.6.0
-jupyter_core==5.7.1
-jupyter_server==2.14.2
-jupyter_server_terminals==0.5.3
-jupyterlab==4.1.6
-jupyterlab_pygments==0.3.0
-jupyterlab_server==2.27.3
-jupytext==1.16.1
-kiwisolver==1.4.5
-langcodes==3.3.0
-lark==1.2.2
-lazy_loader==0.3
-librosa==0.10.1
-lm-format-enforcer==0.10.6
-Markdown==3.5.2
-markdown-it-py==3.0.0
-matplotlib==3.8.2
-matplotlib-inline==0.1.6
-mdit-py-plugins==0.4.0
-mdurl==0.1.2
-mistral_common==1.3.4
-mistune==3.0.2
-mkl==2021.1.1
-mkl-devel==2021.1.1
-mkl-include==2021.1.1
-mock==5.1.0
-mpmath==1.3.0
-msgpack==1.0.7
-msgspec==0.18.6
-multiprocess==0.70.16
-murmurhash==1.0.10
-nbclient==0.9.0
-nbconvert==7.16.0
-nbformat==5.9.2
-nest-asyncio==1.6.0
-networkx==2.6.3
-ninja==1.11.1.1
-nltk==3.9.1
-notebook==6.4.10
-notebook_shim==0.2.4
numpy==1.24.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==9.1.0.70
-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-dali-cuda120==1.34.0
-nvidia-ml-py==12.560.30
-nvidia-nccl-cu12==2.20.5
-nvidia-nvjitlink-cu12==12.6.68
-nvidia-nvtx-cu12==12.1.105
-nvidia-pyindex==1.0.9
-nvitop==1.5.1
-oauthlib==3.2.2
-openai==1.43.0
-optree==0.10.0
-orjson==3.10.7
-overrides==7.7.0
-packaging==23.2
pandas==2.2.2
-pandocfilters==1.5.1
-parso==0.8.3
-partial-json-parser==0.2.1.1.post4
peft==0.12.0
-pexpect==4.9.0
-platformdirs==4.2.0
-pluggy==1.4.0
-pooch==1.8.0
-preshed==3.0.9
-prettytable==3.9.0
-prometheus-client==0.19.0
-prometheus-fastapi-instrumentator==7.0.0
-prompt-toolkit==3.0.43
-protobuf==4.24.4
-ptyprocess==0.7.0
-pure-eval==0.2.2
-py-cpuinfo==9.0.0
-pyarrow==17.0.0
-pyasn1==0.5.1
-pyasn1-modules==0.3.0
-pybind11==2.11.1
-pybind11-global==2.11.1
-pycountry==24.6.1
-pycparser==2.21
-pydantic==2.8.2
-pydantic_core==2.20.1
-pydub==0.25.1
-Pygments==2.17.2
-PyJWT==2.8.0
-pyparsing==3.1.1
-pytest==8.0.0
-pytest-flakefinder==1.1.0
-pytest-rerunfailures==13.0
-pytest-shard==0.1.2
-pytest-xdist==3.5.0
-python-dateutil==2.8.2
-python-dotenv==1.0.1
-python-hostlist==1.23.0
-python-json-logger==2.0.7
-python-multipart==0.0.9
-pytorch-quantization==2.1.2
-PyYAML==6.0.1
-pyzmq==25.1.2
-ray==2.35.0
-referencing==0.33.0
+psutil
regex==2023.12.25
-requests==2.32.3
-requests-oauthlib==1.3.1
-rfc3339-validator==0.1.4
-rfc3986-validator==0.1.1
-rouge-chinese==1.0.3
-rpds-py==0.17.1
-rsa==4.9
-ruff==0.6.3
-safetensors==0.4.4
-semantic-version==2.10.0
-Send2Trash==1.8.2
-sentencepiece==0.2.0
-shellingham==1.5.4
-shtab==1.7.1
-six==1.16.0
-smart-open==6.4.0
-sniffio==1.3.1
-sortedcontainers==2.4.0
-soundfile==0.12.1
-soupsieve==2.5
-soxr==0.3.7
-spacy==3.7.2
-spacy-legacy==3.0.12
-spacy-loggers==1.0.5
-sphinx_glpi_theme==0.6
-srsly==2.4.8
-sse-starlette==2.1.3
-stack-data==0.6.3
-starlette==0.38.4
+scipy
sympy==1.12
-tabulate==0.9.0
-tbb==2021.11.0
-tensorboard==2.9.0
-tensorboard-data-server==0.6.1
-tensorboard-plugin-wit==1.8.1
-termcolor==2.4.0
-terminado==0.18.0
-thinc==8.2.3
-threadpoolctl==3.2.0
tiktoken==0.7.0
-tinycss2==1.2.1
-tokenizers==0.19.1
-toml==0.10.2
-tomli==2.0.1
-tomlkit==0.12.0
torch==2.4.0
-torchvision==0.19.0
-tornado==6.4
tqdm==4.66.5
-traitlets==5.9.0
transformers==4.44.2
-triton==3.0.0
-trl==0.9.6
-typer==0.12.5
-types-dataclasses==0.6.6
-types-python-dateutil==2.9.0.20240821
-typing_extensions==4.12.2
-tyro==0.8.10
-tzdata==2024.1
-uri-template==1.3.0
-urllib3==2.2.2
-uvicorn==0.30.6
-uvloop==0.20.0
-vllm==0.6.0
-vllm-flash-attn==2.6.1
-wasabi==1.1.2
-watchfiles==0.24.0
-wcwidth==0.2.13
-weasel==0.3.4
-webcolors==24.8.0
-webencodings==0.5.1
-websocket-client==1.8.0
-websockets==12.0
-Werkzeug==3.0.1
-xdoctest==1.0.2
-xformers==0.0.27.post2
-xxhash==3.5.0
+vllm>=0.6.2
+wandb
+openai==1.43.0
+huggingface-hub
+loguru
+multiprocess==0.70.16
+pebble
+python-dateutil==2.8.2
+matplotlib==3.8.2
+sglang
+timeout-decorator
+func-timeout
+antlr4-python3-runtime==4.11.1
+latex2sympy2
+word2number
+pyarrow==17.0.0
+sentencepiece==0.2.0
+