From d493b7e566f7ca2d67cfc699c779221be32cd9b7 Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Mon, 8 Sep 2025 16:06:27 -0300 Subject: [PATCH 1/3] Fix new arg in validation method Signed-off-by: Flavia Beo --- scripts/generate_layers_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/generate_layers_metrics.py b/scripts/generate_layers_metrics.py index d3245123..4404263f 100644 --- a/scripts/generate_layers_metrics.py +++ b/scripts/generate_layers_metrics.py @@ -473,7 +473,7 @@ def generate_layers_metrics(model_path, batch_size, seq_length, max_new_tokens): cos_sim = tensor_cos_sim(tensor_cpu_out, cuda_output) prefix = get_default_validation_prefix( - model_path, max_new_token, batch_size, seq_length, "float16" + model_path, max_new_token, batch_size, seq_length, "float16", "" ) layer_name = str(layer_key).replace("[", "").replace("]", "") From af29bc2636d2519ad770253dd99f92c07a869d1b Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Mon, 8 Sep 2025 16:18:39 -0300 Subject: [PATCH 2/3] Fix second new arg in validation method Signed-off-by: Flavia Beo --- scripts/generate_layers_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/generate_layers_metrics.py b/scripts/generate_layers_metrics.py index 4404263f..80d151b6 100644 --- a/scripts/generate_layers_metrics.py +++ b/scripts/generate_layers_metrics.py @@ -473,7 +473,7 @@ def generate_layers_metrics(model_path, batch_size, seq_length, max_new_tokens): cos_sim = tensor_cos_sim(tensor_cpu_out, cuda_output) prefix = get_default_validation_prefix( - model_path, max_new_token, batch_size, seq_length, "float16", "" + model_path, max_new_token, batch_size, seq_length, "float16", "", "v0.3.0" ) layer_name = str(layer_key).replace("[", "").replace("]", "") From 9e2cce3a9c9f0d899bf46c1ead460974732db074 Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Tue, 9 Sep 2025 16:20:32 -0300 Subject: [PATCH 3/3] Fix linting Signed-off-by: Flavia Beo --- scripts/generate_layers_metrics.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scripts/generate_layers_metrics.py b/scripts/generate_layers_metrics.py index 80d151b6..b556f471 100644 --- a/scripts/generate_layers_metrics.py +++ b/scripts/generate_layers_metrics.py @@ -473,7 +473,13 @@ def generate_layers_metrics(model_path, batch_size, seq_length, max_new_tokens): cos_sim = tensor_cos_sim(tensor_cpu_out, cuda_output) prefix = get_default_validation_prefix( - model_path, max_new_token, batch_size, seq_length, "float16", "", "v0.3.0" + model_path, + max_new_token, + batch_size, + seq_length, + "float16", + "", + "v0.3.0", ) layer_name = str(layer_key).replace("[", "").replace("]", "")