vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

use std::chrono::system_clock::now() for random seed

Browse source
This commit is contained in:
David Renshaw 2024-04-27 20:37:12 -04:00
parent 4dba7e8114
commit 6134134080
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
common/sampling.cpp
View file

@ -1,5 +1,6 @@
#define LLAMA_API_INTERNAL #define LLAMA_API_INTERNAL
#include "sampling.h" #include "sampling.h"
#include <chrono>
#include <random> #include <random>
struct llama_sampling_context * llama_sampling_init(const struct llama_sampling_params & params) { struct llama_sampling_context * llama_sampling_init(const struct llama_sampling_params & params) {
@ -68,7 +69,7 @@ void llama_sampling_reset(llama_sampling_context * ctx) {
void llama_sampling_set_rng_seed(struct llama_sampling_context * ctx, uint32_t seed) { void llama_sampling_set_rng_seed(struct llama_sampling_context * ctx, uint32_t seed) {
if (seed == LLAMA_DEFAULT_SEED) { if (seed == LLAMA_DEFAULT_SEED) {
seed = time(NULL); seed = static_cast<uint32_t>(std::chrono::system_clock::now().time_since_epoch().count());
} }
ctx->rng.seed(seed); ctx->rng.seed(seed);
} }