Update requirements.txt

requirements.txt

@@ -1,116 +1,3 @@
-accelerate==1.2.1
-addict==2.4.0
-aiofiles==23.2.1
-aiohappyeyeballs==2.4.4
-aiohttp==3.11.11
-aiosignal==1.3.2
-annotated-types==0.7.0
-anyio==4.7.0
-async-timeout==5.0.1
-attrs==24.3.0
-Brotli
-certifi
-charset-normalizer
-click==8.1.8
-dill==0.3.8
-distro==1.9.0
-einops==0.8.0
-exceptiongroup==1.2.2
-fastapi==0.115.6
-ffmpy==0.5.0
-filelock 
-fire==0.7.0
-frozenlist==1.5.0
-fsspec==2024.3.1
-gmpy2 
-gradio==5.9.1
-gradio_client==1.5.2
-grpcio==1.67.1
-h11==0.14.0
-httpcore==1.0.7
-httpx==0.28.1
-huggingface-hub==0.27.0
-idna 
-Jinja2 
-jiter==0.8.2
-lmdeploy==0.5.3
-markdown-it-py==3.0.0
-MarkupSafe 
-mdurl==0.1.2
-mkl-service==2.4.0
-mkl_fft 
-mkl_random 
-mmengine-lite==0.10.5
-mpmath 
-multidict==6.1.0
-multiprocess==0.70.16
-networkx 
-numpy==1.26.4
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-curand-cu12==10.3.7.77
-nvidia-ml-py==12.560.30
-nvidia-nccl-cu12==2.23.4
-openai==1.40.3
-orjson==3.10.12
-packaging==24.2
-pandas==2.2.3
-peft==0.11.1
-pillow 
-platformdirs==4.3.6
-propcache==0.2.1
-protobuf==5.29.2
-psutil==6.1.1
-pyarrow==18.1.0
-pyarrow-hotfix==0.6
-pybind11==2.13.6
-pydantic==2.10.4
-pydantic_core==2.27.2
-pydub==0.25.1
-Pygments==2.18.0
-pynvml==12.0.0
-PySocks 
-python-dateutil==2.9.0.post0
-python-multipart==0.0.20
-python-rapidjson==1.20
-pytz==2024.2
-PyYAML 
-regex==2024.11.6
-requests 
-rich==13.9.4
-ruff==0.8.4
-safehttpx==0.1.6
-safetensors==0.4.5
-semantic-version==2.10.0
-sentencepiece==0.2.0
-shellingham==1.5.4
-shortuuid==1.0.13
-six==1.17.0
-sniffio==1.3.1
-starlette==0.41.3
-sympy 
-tiktoken==0.8.0
-timm==1.0.8
-tokenizers==0.21.0
-tomli==2.2.1
-tomlkit==0.13.2
-torch==2.1.2
-torchaudio==2.1.2
-torchvision==0.16.2
-tqdm==4.67.1
-transformers==4.47.1
-transformers-stream-generator==0.0.5
-triton==2.1.0
-tritonclient==2.53.0
-typer==0.15.1
-typing_extensions 
-tzdata==2024.2
-urllib3 
-uvicorn==0.34.0
-websockets==14.1
-xxhash==3.5.0
-yapf==0.43.0
-yarl==1.18.3
 torch==2.1.2
 torchvision==0.16.2
 torchaudio==2.1.2
@@ -118,16 +5,35 @@ termcolor==2.4.0
 streamlit==1.39.0
 class_registry==2.1.2
 datasets==3.1.0
-griffe==0.48.0
-asyncer
+# -r requirements/optional.txt
+google-search-results
+lmdeploy>=0.2.5
+pillow
+python-pptx
+timeout_decorator
+torch
+transformers>=4.34,<=4.40
+vllm>=0.3.3
+# -r requirements/runtime.txt
+aiohttp
 arxiv
 asyncache
+asyncer
+distro
 duckduckgo_search==5.3.1b1
+filelock
 func_timeout
+griffe<1.0
 json5
 jsonschema
 jupyter==1.0.0
 jupyter_client==8.6.2
 jupyter_core==5.7.2
+pydantic==2.6.4
+requests
+termcolor
+tiktoken
 timeout-decorator
 typing-extensions
+griffe==0.48.0
+annotated-types==0.7.0