Dependencies mismatched

[siddhantoon]

I followed the setup giude in the readme. I am using a uv managed venv.

It requires to install pytorch 2.1.2, but when you run the main.py it somewhere needs flash_attn
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121

Flash attn requires pytorch>=2.2

Other than this these are not present in requirements.txt but are needed while running,

jupyter
deepseed
matplotlib