flake : fix

hazelnutcloud · Mar 10, 2024 · 69aedef · 69aedef
1 parent b282f87
commit 69aedef
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
@@ -36,8 +36,10 @@ class SentencePieceTokenTypes(IntEnum):
     UNUSED = 5
     BYTE = 6
 
+
 AnyModel = TypeVar("AnyModel", bound="type[Model]")
 
+
 class Model(ABC):
     _model_classes: dict[str, type[Model]] = {}
 
@@ -187,6 +189,7 @@ def load_hparams(dir_model):
     @classmethod
     def register(cls, *names: str) -> Callable[[AnyModel], AnyModel]:
         assert names
+
         def func(modelcls: type[Model]):
             for name in names:
                 cls._model_classes[name] = modelcls