Skip to content

Commit 6b990d2

Browse files
pcullitoncopybara-github
authored andcommitted
Internal change.
PiperOrigin-RevId: 732288967
1 parent 2bdf26d commit 6b990d2

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

compression/shared.h

+7-2
Original file line numberDiff line numberDiff line change
@@ -186,11 +186,16 @@ constexpr bool IsNuqStream() {
186186
}
187187

188188
// Instruction-tuned models require extra 'turn structure' tokens in prompts.
189-
enum class PromptWrapping { GEMMA_IT, GEMMA_PT, PALIGEMMA };
189+
enum class PromptWrapping {
190+
GEMMA_IT,
191+
GEMMA_PT,
192+
PALIGEMMA,
193+
kSentinel // must be last
194+
};
190195

191196
inline bool EnumValid(PromptWrapping type) {
192197
return static_cast<int>(type) >= 0 &&
193-
static_cast<int>(type) <= static_cast<int>(PromptWrapping::PALIGEMMA);
198+
static_cast<int>(type) < static_cast<int>(PromptWrapping::kSentinel);
194199
}
195200

196201
// Tensor types for loading weights. Note that not all types are supported as

0 commit comments

Comments
 (0)