Revert "whoops"

This reverts commit 25f74c3.
pytorch · Feb 13, 2025 · d2a3e65 · d2a3e65
1 parent 25f74c3
commit d2a3e65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/recipes/test_full_finetune_distributed.py b/tests/recipes/test_full_finetune_distributed.py
@@ -267,6 +267,8 @@ def test_loss_single_rank(
         "config, model_type, ckpt_type, micro_batch_size, gradient_accumulation_steps, optim_in_bwd",
         [
             ("llama3/8B_full", "llama3", "tune", 1, 4, False),
+        ],
+        [
             ("llama3/8B_full", "llama3", "tune", 4, 1, True),
         ],
     )