add back qat test case

ebsmothers · ebsmothers · commit 83cba272ddd4 · 2024-10-31T07:06:42.000-07:00
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),
         ],

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@ def _fetch_expected_loss_values(self, model_type):`
`54`	`54`	`@pytest.mark.parametrize(`
`55`	`55`	`"config, model_type, ckpt_type, micro_batch_size, gradient_accumulation_steps",`
`56`	`56`	`[`
`57`		`- # ("llama2/7B_qat_full", "llama2", "hf", 4, 1),`
	`57`	`+ ("llama2/7B_qat_full", "llama2", "hf", 4, 1),`
`58`	`58`	`("llama3/8B_qat_full", "llama3", "tune", 4, 1),`
`59`	`59`	`("llama3/8B_qat_full", "llama3", "tune", 1, 4),`
`60`	`60`	`],`