From d2b77cce91f5d74ca7b4761cf1555009cfb5ec81 Mon Sep 17 00:00:00 2001 From: Douglas Hanley Date: Thu, 15 Feb 2024 09:51:16 -0600 Subject: [PATCH] small typing fix from linter --- gguf-py/gguf/gguf_writer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index 9d8d7f924..385e65cc1 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -19,6 +19,7 @@ from .constants import ( GGUFValueType, Keys, RopeScalingType, + PoolingType, TokenType, )