From 1d18c5a1292ef40c6c37ca83dd4595bbc15cdc84 Mon Sep 17 00:00:00 2001 From: Paul Chang Date: Mon, 29 Apr 2024 06:49:10 -0700 Subject: [PATCH] Improve documentation for compress_weights flags PiperOrigin-RevId: 629053191 --- gemma/compress_weights.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gemma/compress_weights.cc b/gemma/compress_weights.cc index a4d414c..d9260b0 100644 --- a/gemma/compress_weights.cc +++ b/gemma/compress_weights.cc @@ -70,7 +70,7 @@ struct Args : public ArgsBase { template void ForEach(const Visitor& visitor) { visitor(weights, "weights", Path(), - "Path name of model weights (.sbs) file.\n" + "Path to model weights (.bin) file.\n" " Required argument."); visitor(model_type_str, "model", std::string(), "Model type\n 2b-it = 2B parameters, instruction-tuned\n " @@ -80,7 +80,7 @@ struct Args : public ArgsBase { "gr2b-pt = griffin 2B parameters, pretrained\n " " Required argument."); visitor(compressed_weights, "compressed_weights", Path(), - "Path name where compressed weights file will be written.\n" + "Path name where compressed weights (.sbs) file will be written.\n" " Required argument."); visitor(num_threads, "num_threads", kDefaultNumThreads, // see ChooseNumThreads