mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 14:52:20 +00:00
feat(backend): use llama_token as TokenId type
This commit is contained in:
parent
45d5a6a8c5
commit
92bb113653
@ -19,7 +19,7 @@ namespace huggingface::tgi::backends::llama {
|
||||
|
||||
|
||||
class TgiLlamaCppBackend {
|
||||
using TokenId = int32_t;
|
||||
using TokenId = llama_token;
|
||||
|
||||
private:
|
||||
llama_model* model;
|
||||
|
Loading…
Reference in New Issue
Block a user