requirements.txt

# --extra-index-url https://download.pytorch.org/whl/cu118
fastapi==0.100.1
pydantic==2.1.1
pylint==2.17.5
uvicorn>=0.23.2
torch==2.0.1
transformers==4.31.0
accelerate==0.21.0
diffusers==0.19.3
torchvision==0.15.2
safetensors==0.3.1
huggingface-hub==0.16.4