We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2bdf26d commit 6b990d2Copy full SHA for 6b990d2
compression/shared.h
@@ -186,11 +186,16 @@ constexpr bool IsNuqStream() {
186
}
187
188
// Instruction-tuned models require extra 'turn structure' tokens in prompts.
189
-enum class PromptWrapping { GEMMA_IT, GEMMA_PT, PALIGEMMA };
+enum class PromptWrapping {
190
+ GEMMA_IT,
191
+ GEMMA_PT,
192
+ PALIGEMMA,
193
+ kSentinel // must be last
194
+};
195
196
inline bool EnumValid(PromptWrapping type) {
197
return static_cast<int>(type) >= 0 &&
- static_cast<int>(type) <= static_cast<int>(PromptWrapping::PALIGEMMA);
198
+ static_cast<int>(type) < static_cast<int>(PromptWrapping::kSentinel);
199
200
201
// Tensor types for loading weights. Note that not all types are supported as
0 commit comments