highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-04-12 13:49:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

sample: add nil check for grammar

Browse Source
This commit is contained in:
ParthSareen 2025-04-03 11:37:51 -07:00
parent 106592820d
commit ad9791da2e
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
sample/samplers.go
View File

@ -171,6 +171,9 @@ type Grammar struct {
func NewGrammar(vocab *model.Vocabulary, grammarStr string) (*Grammar, error) {
grammar := llama.InitGrammarChain(grammarStr)
if grammar == nil {
return nil, errors.New("sample: failed to initialize grammar")
}
for _, s := range vocab.Values {
id := vocab.Encode(s)
grammar.AddSymbol(s, uint32(id))