Lint

Smartappli · web-flow · commit e211d7bde945 · 2024-08-02T18:50:59.000+02:00
diff --git a/llama_cpp/llama_tokenizer.py b/llama_cpp/llama_tokenizer.py
@@ -2,9 +2,9 @@
 
 import abc
 from typing import (
+    Any,
     List,
     Optional,
-    Any,
 )
 
 import llama_cpp
@@ -62,7 +62,7 @@ def decode(self, tokens: List[int]) -> str:
         return self.detokenize(tokens).decode("utf-8", errors="ignore")
 
     @classmethod
-    def from_ggml_file(cls, path: str) -> "LlamaTokenizer":
+    def from_ggml_file(cls, path: str) -> LlamaTokenizer:
         return cls(llama_cpp.Llama(model_path=path, vocab_only=True))
 
 
@@ -92,7 +92,7 @@ def detokenize(
             return self.hf_tokenizer.decode(tokens).encode("utf-8", errors="ignore")
 
     @classmethod
-    def from_pretrained(cls, pretrained_model_name_or_path: str) -> "LlamaHFTokenizer":
+    def from_pretrained(cls, pretrained_model_name_or_path: str) -> LlamaHFTokenizer:
         try:
             from transformers import AutoTokenizer
         except ImportError:
