From 3708bf9e9e7379fe17cbdd8bb4d2367f5c6a3d99 Mon Sep 17 00:00:00 2001 From: Eugen Hotaj Date: Sun, 19 Jan 2025 16:17:49 -0800 Subject: [PATCH] Update missed test, fix miscopied values. --- tests/recipes/test_lora_finetune_distributed.py | 2 +- tests/recipes/test_qat_distributed.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/recipes/test_lora_finetune_distributed.py b/tests/recipes/test_lora_finetune_distributed.py index 790bc780b3..6ea05c3aa0 100644 --- 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 index 34dd190125..9f2193a13f 100644 --- 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]