revert change to recipe test

Author: ebsmothers

tests/recipes/test_lora_finetune_single_device.py

@@ -119,6 +119,10 @@ def test_loss_qlora(self, compile, dtype, tmpdir, monkeypatch):
         ckpt_dir = ckpt_path.parent
         log_file = gen_log_file_name(tmpdir)
 
+        # To workaround https://github.com/pytorch/torchtune/issues/676
+        if compile:
+            os.environ["TORCH_COMPILE_BACKEND"] = "aot_eager"
+
         cmd = f"""
         tune run lora_finetune_single_device
             --config llama2/7B_qlora_single_device \
@@ -134,12 +138,6 @@ def test_loss_qlora(self, compile, dtype, tmpdir, monkeypatch):
             compile={compile} \
         """.split()
 
-        if compile:
-            os.environ["TORCH_COMPILE_BACKEND"] = "aot_eager"
-            if not torch_version_ge("2.5.0"):
-                # Chunked cross entropy not support on 2.4
-                cmd = cmd + ["loss=torch.nn.CrossEntropyLoss"]
-
         model_config = MODEL_TEST_CONFIGS["llama2_qlora"]
 
         cmd = cmd + self._get_test_config_overrides(dtype_str=dtype) + model_config