happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve documentation for compress_weights flags 

PiperOrigin-RevId: 629053191
This commit is contained in:
Paul Chang 2024-04-29 06:49:10 -07:00 committed by Copybara-Service
parent 7a12e29027
commit 1d18c5a129
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gemma/compress_weights.cc
View File

@ -70,7 +70,7 @@ struct Args : public ArgsBase<Args> {
template <class Visitor> template <class Visitor>
void ForEach(const Visitor& visitor) { void ForEach(const Visitor& visitor) {
visitor(weights, "weights", Path(), visitor(weights, "weights", Path(),
"Path name of model weights (.sbs) file.\n" "Path to model weights (.bin) file.\n"
" Required argument."); " Required argument.");
visitor(model_type_str, "model", std::string(), visitor(model_type_str, "model", std::string(),
"Model type\n 2b-it = 2B parameters, instruction-tuned\n " "Model type\n 2b-it = 2B parameters, instruction-tuned\n "
@ -80,7 +80,7 @@ struct Args : public ArgsBase<Args> {
"gr2b-pt = griffin 2B parameters, pretrained\n " "gr2b-pt = griffin 2B parameters, pretrained\n "
" Required argument."); " Required argument.");
visitor(compressed_weights, "compressed_weights", Path(), 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."); " Required argument.");
visitor(num_threads, "num_threads", visitor(num_threads, "num_threads",
kDefaultNumThreads, // see ChooseNumThreads kDefaultNumThreads, // see ChooseNumThreads