diff --git a/tools/modelwriter.h b/tools/modelwriter.h index 88ccb948a9c..4f445cfe2a4 100644 --- a/tools/modelwriter.h +++ b/tools/modelwriter.h @@ -2007,6 +2007,7 @@ int ModelWriter::save(const char* parampath, const char* binpath) fprintf_param_value(" 3=%d", kdim) fprintf_param_value(" 4=%d", vdim) fprintf_param_value(" 5=%d", attn_mask) + fprintf_param_value(" 6=%e", scale) fwrite_weight_tag_data(op->q_weight_data, bp); fwrite_weight_data(op->q_bias_data, bp);