Fix nightly tests for qat_lora_fintune_distributed

[andrewor14]

Summmary: These were not caught in CI because the tests require torchao 0.7+. Here is an example of the failure: https://github.com/pytorch/torchtune/actions/runs/12037600747/job/33561415524

Test Plan:
(on local devgpu)

(pytorch-3.10) andrewor@devgpu022:~/local/torchtune $ pytest -m integration_test tests/recipes/test_qat_lora_finetune_distributed.py
Expected artifacts for test run are:
small-ckpt-tune-03082024.pt
small-ckpt-meta-03082024.pt
small-ckpt-hf-03082024.pt
small-ckpt-tune-llama3-05052024.pt
small-ckpt-hf-reward-07122024.pt
small-ckpt-meta-vision-10172024.pt
small-ckpt-hf-vision-10172024.pt
tokenizer.model
tokenizer_llama3.model
File already exists locally: /tmp/test-artifacts/small-ckpt-tune-03082024.pt
File already exists locally: /tmp/test-artifacts/small-ckpt-meta-03082024.pt
File already exists locally: /tmp/test-artifacts/small-ckpt-hf-03082024.pt
File already exists locally: /tmp/test-artifacts/small-ckpt-tune-llama3-05052024.pt
File already exists locally: /tmp/test-artifacts/small-ckpt-hf-reward-07122024.pt
File already exists locally: /tmp/test-artifacts/small-ckpt-meta-vision-10172024.pt
File already exists locally: /tmp/test-artifacts/small-ckpt-hf-vision-10172024.pt
File already exists locally: /tmp/test-artifacts/tokenizer.model
File already exists locally: /tmp/test-artifacts/tokenizer_llama3.model
=================================================================================================== test session starts ====================================================================================================
platform linux -- Python 3.10.13, pytest-7.4.0, pluggy-1.4.0
rootdir: /data/users/andrewor/torchtune
configfile: pyproject.toml
plugins: hypothesis-6.87.3, anyio-4.2.0, integration-0.2.3, xdist-3.6.1, mock-3.14.0, cov-5.0.0
collected 4 items                                                                                                                                                                                                          

tests/recipes/test_qat_lora_finetune_distributed.py ....                                                                                                                                                             [100%]

=============================================================================================== 4 passed in 75.60s (0:01:15) ===============================================================================================

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchtune/2085

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit b47896d with merge base 437a8ff ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[ebsmothers]

Thank you for the fix!