Skip to content

Commit 2e2dd07

Browse files
Fix YAML seed parsing to handle negative values properly
- Convert negative seed values (-1) to LLAMA_DEFAULT_SEED - Allows YAML config files to use seed: -1 as expected - Fixes parsing error in example config files Co-Authored-By: Jaime Mizrachi <[email protected]>
1 parent d1e7175 commit 2e2dd07

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

common/arg.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,8 @@ static bool common_params_load_from_yaml(const std::string & config_file, common
141141

142142
// Sampling parameters
143143
if (config["seed"]) {
144-
params.sampling.seed = config["seed"].as<uint32_t>();
144+
int32_t seed_val = config["seed"].as<int32_t>();
145+
params.sampling.seed = (seed_val == -1) ? LLAMA_DEFAULT_SEED : static_cast<uint32_t>(seed_val);
145146
}
146147
if (config["temperature"]) {
147148
params.sampling.temp = config["temperature"].as<float>();

0 commit comments

Comments
 (0)