Setup broken with PyTorch 2.5.x

[juliensimon]

Running setup.sh fails with undefined symbol: __nvJitLinkAddData_12_1, version libnvJitLink.so.12.

This isn't a DistillKit issue, but a known problem in PyTorch: see pytorch/pytorch#111469.

Until they fix it, pinning torch to 2.4.0 (torch==2.4) solves the problem :)