Update missed test, fix miscopied values.

pytorch · Jan 20, 2025 · 3708bf9 · 3708bf9
1 parent aec6477
commit 3708bf9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/recipes/test_lora_finetune_distributed.py b/tests/recipes/test_lora_finetune_distributed.py
@@ -55,7 +55,7 @@ def _fetch_expected_loss_values(self, model_type):
         # https://gist.github.com/ebsmothers/f1c3db7c66655a23a91e0290360960c4
         loss_values_map = {
             "llama2": [10.5320, 10.5608, 10.4895, 10.5068],
-            "llama3": [11.9259, 11.9250, 11.9753, 11.9774],
+            "llama3": [11.9265, 11.9255, 11.9754, 11.9780],
         }
         return loss_values_map[model_type]
 

diff --git a/tests/recipes/test_qat_distributed.py b/tests/recipes/test_qat_distributed.py
@@ -45,8 +45,8 @@ def _get_test_config_overrides(self):
 
     def _fetch_expected_loss_values(self, model_type):
         loss_values_map = {
-            "llama2": [10.5211, 10.5217, 10.4944, 10.5134],
-            "llama3": [11.9836, 11.9683, 11.9594, 11.9366],
+            "llama2": [10.5337, 10.5563, 10.4786, 10.5002],
+            "llama3": [11.9270, 11.9240, 11.9731, 11.9751],
         }
         return loss_values_map[model_type]