SegFly-Firefly / requirements.txt
markus-42's picture
Initial commit
94cbfa6
Raw
History Blame Contribute Delete
229 Bytes
torch>=2.9
transformers==4.37.2
evaluate>=0.4.0
datasets>=2.20.0
numpy>=1.24.4
pandas>=1.5.3
albumentations>=1.3.1
opencv-python-headless>=4.9.0
safetensors
tqdm
pyyaml
matplotlib>=3.8.2
torchmetrics>=1.8.2
torchvision
termcolor