should not be static
This commit is contained in:
parent
43949f7c7c
commit
1abcdb2394
2 changed files with 2 additions and 2 deletions
|
@ -426,7 +426,7 @@ gpt_vocab::id gpt_sample_top_k_top_p(
|
||||||
return logits_id[idx].second;
|
return logits_id[idx].second;
|
||||||
}
|
}
|
||||||
|
|
||||||
static bool should_transpose_layer(std::string name)
|
bool should_transpose_layer(std::string name)
|
||||||
{
|
{
|
||||||
|
|
||||||
if(name.find(".mlp.fc_in.weight")!=std::string::npos ||
|
if(name.find(".mlp.fc_in.weight")!=std::string::npos ||
|
||||||
|
|
|
@ -76,4 +76,4 @@ bool utils_gpt_params_parse(int argc, char ** argv, gpt_params & params);
|
||||||
void utils_gpt_print_usage(int argc, char ** argv, const gpt_params & params);
|
void utils_gpt_print_usage(int argc, char ** argv, const gpt_params & params);
|
||||||
std::string utils_gpt_random_prompt(std::mt19937 & rng);
|
std::string utils_gpt_random_prompt(std::mt19937 & rng);
|
||||||
|
|
||||||
static bool should_transpose_layer(std::string name);
|
bool should_transpose_layer(std::string name);
|
Loading…
Add table
Add a link
Reference in a new issue