# CPU-only dependencies for local development
# torch==2.3.1  # CPU version of PyTorch
# conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 -c pytorch

# GPU dependencies for server (use GPU-specific versions)
# You can comment these out for local dev, and uncomment them on the server.
# torch==2.3.1+cu121  # GPU version of PyTorch
# torchaudio==2.4.1+cu121  # GPU version of torchaudio
# torchvision==0.18.1+cu121  # GPU version of torchvision

# CUDA 12.1
# conda install pytorch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 pytorch-cuda=12.1 -c pytorch -c nvidia


# Other dependencies that are common to both environments
accelerate==1.1.1
aiohappyeyeballs==2.4.3
aiohttp==3.11.5
aiosignal==1.3.1
annotated-types==0.7.0
anyio==4.6.2.post1
attrs==24.2.0
click==8.1.7
cloudpickle==3.1.0
cmake==3.31.0.1
datasets==3.1.0
dill==0.3.8
diskcache==5.6.3
distro==1.9.0
fastapi==0.115.5
frozenlist==1.5.0
fsspec==2024.9.0
gmpy2==2.1.0
h11==0.14.0
hpack==4.0.0
httpcore==1.0.7
httptools==0.6.4
httpx==0.27.2
huggingface-hub==0.26.2
idna==3.4
interegular==0.3.3
Jinja2==3.1.2
jiter==0.7.1
joblib==1.4.2
jsonschema==4.23.0
jsonschema-specifications==2024.10.1
lark==1.2.2
llvmlite==0.43.0
lm-format-enforcer==0.10.3
msgpack==1.1.0
multidict==6.1.0
multiprocess==0.70.16
nest-asyncio==1.6.0
networkx==2.8.4
ninja==1.11.1.1
nltk==3.9.1
numba==0.60.0
numpy==1.26.4
openai==1.54.5
outlines==0.0.46
packaging==24.2
pandas==2.2.3
prometheus-fastapi-instrumentator==7.0.0
prometheus_client==0.21.0
propcache==0.2.0
protobuf==5.28.3
psutil==6.1.0
py-cpuinfo==9.0.0
pyairports==2.1.1
pyarrow==18.0.0
pycountry==24.6.1
pydantic==2.9.2
pydantic_core==2.23.4
python-dateutil==2.9.0.post0
python-dotenv==1.0.1
pytz==2024.2
PyYAML==6.0
pyzmq==26.2.0
ray==2.39.0
referencing==0.35.1
regex==2024.11.6
rpds-py==0.21.0
safetensors==0.4.5
sentencepiece==0.2.0
six==1.16.0
sniffio==1.3.1
starlette==0.41.3
tiktoken==0.8.0
tokenizers==0.20.3
tqdm==4.67.0
transformers==4.46.3
# triton==2.3.1
tzdata==2024.2
uvicorn==0.32.0
uvloop==0.21.0
vllm==0.5.3.post1
vllm-flash-attn==2.5.9.post1
watchfiles==0.24.0
websockets==14.1
xformers==0.0.27
xxhash==3.5.0
yarl==1.17.2
zstandard==0.23.0
