gptmodel 1.0.6__tar.gz → 1.0.8__tar.gz

{gptmodel-1.0.6 → gptmodel-1.0.8}/PKG-INFO

@@ -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.6 → gptmodel-1.0.8}/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 → gptmodel-1.0.8}/gptmodel.egg-info/PKG-INFO

@@ -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/gptmodel.egg-info/requires.txt

@@ -0,0 +1,4 @@
+torch
+numpy
+tiktoken
+tqdm

{gptmodel-1.0.6 → gptmodel-1.0.8}/setup.py

@@ -5,13 +5,13 @@
 # --------------------------> A SAPIENS TECHNOLOGY®️ PRODUCTION) <--------------------------
 from setuptools import setup, find_packages
 package_name = 'gptmodel'
-version = '1.0.6'
+version = '1.0.8'
 setup(
     name=package_name,
     version=version,
     author='SAPIENS TECHNOLOGY',
     packages=find_packages(),
-    install_requires=['torch==2.4.1', 'numpy==1.25.2', 'tiktoken==0.4.0', 'tqdm==4.67.1'],
+    install_requires=['torch', 'numpy', 'tiktoken', 'tqdm'],
     url='https://github.com/sapiens-technology/gptmodel',
     license='Proprietary Software'
 )

gptmodel-1.0.6/gptmodel.egg-info/requires.txt

@@ -1,4 +0,0 @@
-torch==2.4.1
-numpy==1.25.2
-tiktoken==0.4.0
-tqdm==4.67.1