From 32e037ffbe7af7be1bb620b6fe64861fe6de2843 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=2E=20Yusuf=20Sar=C4=B1g=C3=B6z?= <yusufsarigoz@gmail.com>
Date: Sun, 30 Jul 2023 01:01:13 +0300
Subject: [PATCH] gguf : fix set is not subscriptable

---
 gguf.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gguf.py b/gguf.py
index 7e15e5a77..3bcc15f8b 100644
--- a/gguf.py
+++ b/gguf.py
@@ -149,7 +149,7 @@ class GGUFWriter:
         elif vtype == GGUFValueType.ARRAY:
             ltype = set([GGUFValueType.get_type(item) for item in val])
             assert len(ltype) == 1, "All items in a GGUF array should be of the same type"
-            self.fout.write(struct.pack("<I", ltype[0]))
+            self.fout.write(struct.pack("<I", list(ltype)[0]))
             self.fout.write(struct.pack("<I", len(val)))
             for item in val:
                 self.write_val(item, write_vtype=False)
@@ -273,7 +273,7 @@ class GGUFWriter:
     def write_token_list(self, tokens: List):
         self.write_array(constants.KEY_TOKENIZER_LIST, tokens)
 
-    def write_token_scores(self, scores: List:
+    def write_token_scores(self, scores: List[float]):
         self.write_array(constants.KEY_TOKENIZER_SCORES, scores)