add back qat test case

pytorch · Oct 31, 2024 · 83cba27 · 83cba27
1 parent a878829
commit 83cba27
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/recipes/test_qat_distributed.py b/tests/recipes/test_qat_distributed.py
@@ -54,7 +54,7 @@ def _fetch_expected_loss_values(self, model_type):
     @pytest.mark.parametrize(
         "config, model_type, ckpt_type, micro_batch_size, gradient_accumulation_steps",
         [
-            # ("llama2/7B_qat_full", "llama2", "hf", 4, 1),
+            ("llama2/7B_qat_full", "llama2", "hf", 4, 1),
             ("llama3/8B_qat_full", "llama3", "tune", 4, 1),
             ("llama3/8B_qat_full", "llama3", "tune", 1, 4),
         ],