requirements.txt

python=3.10
pytorch==2.1.2
torchvision==0.16.2
torchaudio==2.1.2
termcolor==2.4.0
pytorch-cuda=12.1
class_registry==2.1.2
datasets==3.1.0
griffe==0.48.0
lagent==0.5.0rc1