RxCodexV1MiniAPI / requirements.txt
rxmha125's picture
Upload 3 files
d95d4a2 verified
raw
history blame contribute delete
306 Bytes
# requirements.txt
fastapi
uvicorn[standard]
transformers
sentencepiece
# Install the CPU-only version of PyTorch for smaller size and faster install
# If you have a powerful NVIDIA GPU, you can remove this line and install the CUDA version
torch --index-url https://download.pytorch.org/whl/cpu