fix vocabulary (#10679)

This commit is contained in:
Michael Yang
2025-05-12 17:29:46 -07:00
committed by GitHub
parent a7240c6d63
commit 526b2ed102
7 changed files with 28 additions and 26 deletions

View File

@@ -49,6 +49,9 @@ func New(c fs.Config) (model.Model, error) {
AddBOS: c.Bool("tokenizer.ggml.add_bos_token", true),
EOS: int32(c.Uint("tokenizer.ggml.eos_token_id")),
AddEOS: c.Bool("tokenizer.ggml.add_eos_token", false),
// TODO: set EOT to EOS otherwise 0 will stop generation
EOT: int32(c.Uint("tokenizer.ggml.eos_token_id")),
AddEOT: c.Bool("tokenizer.ggml.add_eos_token", false),
},
),
ImageProcessor: newImageProcessor(c),