foundation-model-stack / fms-hf-tuning

🚀 Collection of tuning recipes with HuggingFace SFTTrainer and PyTorch FSDP.
Apache License 2.0
28 stars 48 forks source link

build(deps): torch<2.5 due to FA2 error with new version #375

Closed anhuong closed 1 month ago

anhuong commented 1 month ago

Description of the change

Reported issue with building FA2 with latest torch 2.5 that came out yesterday morning. Thus will cap torch below 2.5.

Related issue number

How to verify the PR

Was the PR tested

github-actions[bot] commented 1 month ago

Thanks for making a pull request! 😃 One of the maintainers will review and advise on the next steps.