Update requirements.txt

requirements.txt

@@ -1,3 +1,116 @@
+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
@@ -5,34 +118,16 @@ termcolor==2.4.0
 streamlit==1.39.0
 class_registry==2.1.2
 datasets==3.1.0
-# -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
+griffe==0.48.0
+asyncer
 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