pytorch-lightning>=2.5.0,<2.5.4
# the below is uncommented when main is targeting a specific pl dev master commit
# git+https://github.com/Lightning-AI/pytorch-lightning.git@878ecf56b06d5ae3f482e146e78accabc685bfc7#egg=pytorch-lightning
torch>=2.2.0
