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

Add review fixes

Browse source
This commit is contained in:
Michael Klimenko 2024-01-28 10:32:27 +01:00
parent 6ba560d4ce
commit c35004b90d
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
common/sampling.cpp
View file

@ -197,7 +197,7 @@ static llama_token llama_sampling_sample_impl(
}
// apply params.logit_bias map
for (auto logit_bia : params.logit_bias) {
for (const auto & logit_bia : params.logit_bias) {
logits[logit_bia.first] += logit_bia.second;
}

8
common/train.cpp
View file

@ -46,21 +46,21 @@ void free_train_state(struct train_state * state) {
struct random_normal_distribution * init_random_normal_distribution(
int seed, float mean, float std, float min, float max
) {
auto rnd = new random_normal_distribution{ std::mt19937(seed), std::normal_distribution<float>{mean, std}, min, max};
auto rnd = new random_normal_distribution{std::mt19937(seed), std::normal_distribution<float>{mean, std}, min, max};
return rnd;
}
struct random_uniform_distribution * init_random_uniform_distribution(int seed, float min, float max) {
auto rnd = new random_uniform_distribution{ std::mt19937(seed), std::uniform_real_distribution<float>{min, max} };
auto rnd = new random_uniform_distribution{std::mt19937(seed), std::uniform_real_distribution<float>{min, max}};
return rnd;
}
void free_random_normal_distribution (struct random_normal_distribution * rnd) {
free(rnd);
delete rnd;
}
void free_random_uniform_distribution(struct random_uniform_distribution * rnd) {
free(rnd);
delete rnd;
}
struct ggml_tensor * randomize_tensor_normal(struct ggml_tensor * tensor, struct random_normal_distribution * rnd) {