gpu_machine_learner / requirements.txt
coder160's picture
first commit
31166da
raw history blame
No virus
319 Bytes
torch
torchvision
torchaudio
invisible_watermark
datasets
hf-doc-builder
huggingface-hub
Jinja2
librosa
numpy
scipy
tensorboard
omegaconf
pytorch-lightning
xformers
fastapi
pydantic
uvicorn
typing
requests
bs4
transformers
transformers[sentencepiece]
diffusers
diffusers[torch]
diffusers[flax]
accelerate
safetensors