gptmodel 1.0.6__py3-none-any.whl → 1.0.8__py3-none-any.whl

gptmodel/gptmodel.py

@@ -340,7 +340,7 @@ class GPTModel:
                 'embedding_dim': max((1, int(self.__embedding_dim))) if type(self.__embedding_dim) in (bool, int, float) else -1,
                 'vocab_size': max((0, int(self.__vocab_size))) if type(self.__vocab_size) in (bool, int, float) else 0,
                 'block_size': max((1, int(self.__block_size))) if type(self.__block_size) in (bool, int, float) else -1,
-                'end_tag': str(self.__end_tag) if type(self.__end_tag) is not None else '',
+                'end_tag': str(self.__end_tag) if self.__end_tag is not None else '',
                 'number_heads': max((1, int(self.__number_heads))) if type(self.__number_heads) in (bool, int, float) else -1,
                 'number_layers': max((1, int(self.__number_layers))) if type(self.__number_layers) in (bool, int, float) else -1,
                 'dropout': max((0, int(self.dropout))) if type(self.dropout) in (bool, int, float) else 0.1,
@@ -375,7 +375,9 @@ class GPTModel:
                 for _ in self.__tqdm(range(10), desc='Loading model', leave=False):
                     try: checkpoint = self.__load(model_file, map_location=self.__device)
                     except: checkpoint = self.__load(model_file)
-            else: checkpoint = self.__load(model_file, map_location=self.__device)
+            else:
+                try: checkpoint = self.__load(model_file, map_location=self.__device)
+                except: checkpoint = self.__load(model_file)
             try: self.__tokenizer = str(checkpoint['tokenizer']).lower().strip()
             except: self.__tokenizer = 'gpt'
             try: self.__embedding_dim = max((1, int(checkpoint['embedding_dim']))) if checkpoint['embedding_dim'] != -1 else None

{gptmodel-1.0.6.dist-info → gptmodel-1.0.8.dist-info}/METADATA

@@ -1,14 +1,14 @@
 Metadata-Version: 2.2
 Name: gptmodel
-Version: 1.0.6
+Version: 1.0.8
 Home-page: https://github.com/sapiens-technology/gptmodel
 Author: SAPIENS TECHNOLOGY
 License: Proprietary Software
 License-File: LICENSE.txt
-Requires-Dist: torch==2.4.1
-Requires-Dist: numpy==1.25.2
-Requires-Dist: tiktoken==0.4.0
-Requires-Dist: tqdm==4.67.1
+Requires-Dist: torch
+Requires-Dist: numpy
+Requires-Dist: tiktoken
+Requires-Dist: tqdm
 Dynamic: author
 Dynamic: home-page
 Dynamic: license

gptmodel-1.0.8.dist-info/RECORD

@@ -0,0 +1,7 @@
+gptmodel/__init__.py,sha256=ifnJnBPHcBZwiHgt_mNp9gd1OJYCCHB3bgml6TmKBIw,1358
+gptmodel/gptmodel.py,sha256=C5S8f-TI6WZxN3BtZbs7ANEeIxu6NbvEzMh070IFQVQ,33584
+gptmodel-1.0.8.dist-info/LICENSE.txt,sha256=WqB2vIA5tH5lqLTr53yT_oy1m0wYfuvCPQKxdDHWimg,115
+gptmodel-1.0.8.dist-info/METADATA,sha256=V6fZe-JFauZ7isHexJ71qQchVe90QAKP6JUX6O8ROrw,354
+gptmodel-1.0.8.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+gptmodel-1.0.8.dist-info/top_level.txt,sha256=585C0QclguIkVPKKPpoeD2FxYAc5n5EAuvJNK4vMeQk,9
+gptmodel-1.0.8.dist-info/RECORD,,

gptmodel-1.0.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-gptmodel/__init__.py,sha256=ifnJnBPHcBZwiHgt_mNp9gd1OJYCCHB3bgml6TmKBIw,1358
-gptmodel/gptmodel.py,sha256=rJqV2Pbk7lRati9sqj4hEHUyqPFfjQu6eWYkq4T2sPw,33508
-gptmodel-1.0.6.dist-info/LICENSE.txt,sha256=WqB2vIA5tH5lqLTr53yT_oy1m0wYfuvCPQKxdDHWimg,115
-gptmodel-1.0.6.dist-info/METADATA,sha256=oUIv3Mn6VM2SgqLQ6xLv6F95nsfOrNBFWQ_nLCK8B-c,384
-gptmodel-1.0.6.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-gptmodel-1.0.6.dist-info/top_level.txt,sha256=585C0QclguIkVPKKPpoeD2FxYAc5n5EAuvJNK4vMeQk,9
-gptmodel-1.0.6.dist-info/RECORD,,