Skip to content

Commit 5855282

Browse files
The gemma.cpp Authorscopybara-github
The gemma.cpp Authors
authored andcommitted
Internal change.
PiperOrigin-RevId: 762106575
1 parent eb8a463 commit 5855282

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

gemma/configs.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -574,7 +574,7 @@ ModelConfig::ModelConfig(const Model model, Type weight,
574574
static Model FindModel(const std::string& specifier) {
575575
Model found_model = Model::UNKNOWN;
576576
ForEachModel([&](Model model) {
577-
const char* prefix = ModelPrefix(model);
577+
const std::string prefix = std::string(ModelPrefix(model)) + "-";
578578
if (specifier.rfind(prefix, 0) == 0) { // Starts with prefix.
579579
// We only expect one match.
580580
HWY_ASSERT_M(found_model == Model::UNKNOWN, specifier.c_str());

gemma/weights.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,8 @@ struct LayerWeightsPtrs {
136136
key_norm_scale(Concat("key_norm", suffix_), tensors),
137137
query_norm_scale(Concat("query_norm", suffix_), tensors),
138138

139-
layer_config(config) {}
139+
layer_config(config) {
140+
}
140141
~LayerWeightsPtrs() = default;
141142

142143
const std::string suffix_;
@@ -261,7 +262,7 @@ struct LayerWeightsPtrs {
261262
func(TENSOR_ARGS(gating_einsum_w, kMaybeRead));
262263
func(TENSOR_ARGS(gating_einsum_w1, kMaybeRead));
263264
func(TENSOR_ARGS(gating_einsum_w2, kMaybeRead));
264-
func(TENSOR_ARGS(linear_w, kMustRead));
265+
func(TENSOR_ARGS(linear_w, kMaybeRead));
265266
func(TENSOR_ARGS(pre_attention_norm_scale, kMustRead));
266267
func(TENSOR_ARGS(pre_ffw_norm_scale, kMustRead));
267268
}

0 commit comments

Comments
 (0)