pycocotools
torch>=1.13.0
torchvision>=0.14.0
scipy
tqdm
numpy
transformers<5.0.0,>4.0.0
peft
polygraphy
rf100vl
pydantic
supervision
matplotlib
pillow-avif-plugin==1.5.5

[metrics]
tensorboard>=2.13.0
wandb

[onnxexport]
onnx
onnxsim
onnx_graphsurgeon
onnxruntime
