Skip to content

Commit 193cf74

Browse files
author
Judd
committed
fix reported model type
1 parent 4bdb48c commit 193cf74

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/gemma.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,7 @@ class ConditionalGeneration : public BaseModelForConditionalGeneration
425425
public:
426426
typedef HeterogeneousModel<BaseConfig, Embedding, RMSNorm> ModelClass;
427427
public:
428-
ConditionalGeneration(const Config &config, const RuntimeConfig &runtime_config, ModelType type = MODEL_TYPE_GEMMA2)
428+
ConditionalGeneration(const Config &config, const RuntimeConfig &runtime_config, ModelType type = MODEL_TYPE_GEMMA3)
429429
: BaseModelForConditionalGeneration(type, config, runtime_config, 4096 * 2), config(config),
430430
sliding_window(config.sliding_window),
431431
sliding_window_pattern(config.sliding_window_pattern)

0 commit comments

Comments
 (0)