TrueFrame / requirements.txt
NeelakshSaxena's picture
Deploy auto GPU fallback + FastAPI /predict
8d017ca verified
raw
history blame contribute delete
532 Bytes
# Core ML
numpy>=1.24.0
python-dotenv
gradio
torch>=2.0.0
torchvision>=0.15.0
torchaudio>=2.0.0
torchcodec
Pillow
# Dependencies
albumentations>=0.5.2
datasets
huggingface_hub
scikit-learn>=1.3.0
scikit-image>=0.21.0
pandas>=2.0.0
matplotlib>=3.7.0
seaborn
transformers==4.36.2
peft
accelerate
diffusers
opencv-python
# Optional fallback for lower-memory GPU execution
bitsandbytes
# M2TR specific
yacs==0.1.8
nbconvert
tensorboard==2.20.0
tqdm==4.67.1
PyYAML==6.0.3
simplejson==3.20.2
fvcore