diff --git a/scripts/generate_layers_metrics.py b/scripts/generate_layers_metrics.py index d3245123..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" + model_path, + max_new_token, + batch_size, + seq_length, + "float16", + "", + "v0.3.0", ) layer_name = str(layer_key).replace("[", "").replace("]", "")