Edit requirements

requirements.txt

@@ -1,80 +1,19 @@
-accelerate==0.26.1
-aioprometheus==23.12.0
-aiosignal==1.3.1
-annotated-types==0.6.0
-anyio==4.2.0
-attrs==23.2.0
-certifi==2024.2.2
-charset-normalizer==3.3.2
-click==8.1.7
-cloudpickle==3.0.0
-diskcache==5.6.3
-exceptiongroup==1.2.0
-fastapi==0.109.0
-filelock==3.13.1
-frozenlist==1.4.1
-fsspec==2023.12.2
-h11==0.14.0
-httptools==0.6.1
-huggingface-hub==0.20.3
-idna==3.6
-interegular==0.3.3
-Jinja2==3.1.3
-joblib==1.3.2
-jsonschema==4.21.1
-jsonschema-specifications==2023.12.1
-lark==1.1.9
-llvmlite==0.42.0
-MarkupSafe==2.1.4
-mpmath==1.3.0
-msgpack==1.0.7
-nest-asyncio==1.6.0
-networkx==3.2.1
-ninja==1.11.1.1
-numba==0.59.0
-numpy==1.26.3
-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.18.1
-nvidia-nvjitlink-cu12==12.3.101
-nvidia-nvtx-cu12==12.1.105
-orjson==3.9.12
-packaging==23.2
-protobuf==4.25.2
-psutil==5.9.8
-pydantic==2.6.0
-pydantic_core==2.16.1
-pynvml==11.5.0
-python-dotenv==1.0.1
-PyYAML==6.0.1
-quantile-python==1.1
+interegular
+jinja2
+lark
+nest_asyncio
+numpy
+cloudpickle
+diskcache
+pydantic>=2.0
+scipy
+torch>=2.1.0
+numba
+joblib
+referencing
+jsonschema
+requests
+transformers
 ray==2.9.0
-referencing==0.33.0
-regex==2023.12.25
-requests==2.31.0
-rpds-py==0.17.1
-safetensors==0.4.2
-scipy==1.12.0
-sentencepiece==0.1.99
-sniffio==1.3.0
-starlette==0.35.1
-sympy==1.12
-tokenizers==0.15.1
-torch==2.1.2
-tqdm==4.66.1
-transformers==4.36.2
-triton==2.1.0
-typing_extensions==4.9.0
-urllib3==2.2.0
-uvicorn==0.27.0.post1
-uvloop==0.19.0
-watchfiles==0.21.0
-websockets==12.0
-xformers==0.0.23.post1
+uvicorn
+fastapi