requirement.txt

python=3.10 -y
pytorch==2.1.2
torchvision==0.16.2
torchaudio==2.1.2
pytorch-cuda=12.1
termcolor==2.4.0
streamlit==1.39.0
class_registry==2.1.2
datasets==3.1.0
griffe==0.48.0
lagent