Update app argument documentation

PiperOrigin-RevId: 739159864
This commit is contained in:
Jan Wassenberg 2025-03-21 06:32:54 -07:00 committed by Copybara-Service
parent 05b1cce9f7
commit 6300c123ee
1 changed files with 5 additions and 8 deletions

View File

@ -205,15 +205,12 @@ struct LoaderArgs : public ArgsBase<LoaderArgs> {
visitor(tokenizer, "tokenizer", Path(), visitor(tokenizer, "tokenizer", Path(),
"Path name of tokenizer model file."); "Path name of tokenizer model file.");
visitor(weights, "weights", Path(), visitor(weights, "weights", Path(),
"Path name of model weights (.sbs) file.\n Required argument."); "Path name of model weights (.sbs) file.\n Required argument.\n");
visitor(compressed_weights, "compressed_weights", Path(), visitor(compressed_weights, "compressed_weights", Path(),
"Alias for --weights."); "Deprecated alias for --weights.");
visitor(model_type_str, "model", std::string(), visitor(
"Model type\n 2b-it = 2B parameters, instruction-tuned\n " model_type_str, "model", std::string(),
"2b-pt = 2B parameters, pretrained\n 7b-it = 7B parameters " "Model type, see common.cc for valid values.\n");
"instruction-tuned\n 7b-pt = 7B parameters, pretrained\n "
"gr2b-it = griffin 2B parameters, instruction-tuned\n "
"gr2b-pt = griffin 2B parameters, pretrained.");
visitor(weight_type_str, "weight_type", std::string("sfp"), visitor(weight_type_str, "weight_type", std::string("sfp"),
"Weight type\n f32 = float, bf16 = bfloat16, sfp = 8-bit SFP."); "Weight type\n f32 = float, bf16 = bfloat16, sfp = 8-bit SFP.");
} }