model-code / clean /image /yermandy /requirements.txt
deepsafe's picture
Add stripped inference-only model code mirror
9e14838 verified
Raw History Blame Contribute Delete
435 Bytes
torch==2.6.0
torchaudio==2.6.0
torchvision==0.21.0
lightning==2.5.0
transformers==4.50.0
tqdm==4.67.1 # progress bar
timm==1.0.14 # torch models
matplotlib==3.10.0 # visualization
seaborn==0.13.2 # visualization
scikit-learn==1.6.1 # metrics
rich==13.9.4 # logging
wandb==0.19.4 # logging
pydantic==2.9.2 # config
ruff==0.9.3 # formatting
fire==0.7.0 # CLI
peft==0.14.0 # parameter-efficient fine-tuning
# ipykernel==6.29.5 # jupyter