Overwriting pytorch version 2.5.1 and fairseq2

Hello, I was wondering if anyone has attempted to train the base model in Colab or has faced a similar issue that I am having outside of Colab?

After installing pytorch 2.5.1 (cu121) and fairseq2 0.3.0rc1 (cu121) and run pip install -e ".[data,eval]", this command overwrites pytorch 2.5.1 --> 2.6.0 and fairseq2 0.3.0rc1 --> 0.4.5. According to the repo, LCM is dependent on fairseq2 0.3.0rc1. Was wondering if there was a workaround or if this is a Colab issue. Thank you!