sapiens-gpt 1.0.2__tar.gz → 2.0.4__tar.gz

{sapiens_gpt-1.0.2 → sapiens_gpt-2.0.4}/PKG-INFO

@@ -1,16 +1,16 @@
 Metadata-Version: 2.2
 Name: sapiens_gpt
-Version: 1.0.2
+Version: 2.0.4
 Home-page: https://github.com/sapiens-technology/sapiens_gpt
 Author: SAPIENS TECHNOLOGY
 License: Proprietary Software
 License-File: LICENSE.txt
-Requires-Dist: torch==2.4.1
-Requires-Dist: torch-xla==2.7.0
-Requires-Dist: numpy==1.25.2
-Requires-Dist: tiktoken==0.4.0
+Requires-Dist: torch
+Requires-Dist: numpy
+Requires-Dist: tiktoken
 Requires-Dist: tqdm
 Requires-Dist: certifi
+Requires-Dist: torch-xla==2.7.0; platform_system == "Linux" and platform_machine == "x86_64"
 Dynamic: author
 Dynamic: home-page
 Dynamic: license

{sapiens_gpt-1.0.2 → sapiens_gpt-2.0.4}/sapiens_gpt/sapiens_gpt.py

@@ -62,6 +62,8 @@ class SapiensGPT:
                 elif self.__xm is not None: self.device = self.__xm.xla_device()
                 elif backends.mps.is_available(): self.device = device('mps')
                 else: self.device = device('cpu')
+            from sys import setrecursionlimit
+            setrecursionlimit(self.epochs+10)
             self.__Dataset = Dataset
             self.__Module = Module
             self.__neural_network = artificial_neural_network
@@ -428,6 +430,8 @@ class SapiensGPT:
             string_precision = f'{precision:.4f}'.ljust(5, '0')
             formatted_string = '{desc}: {percentage:3.0f}%|{bar:10}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}, {rate_fmt:>9}]'
             if self.__xm is not None or str(self.device).lower().strip() == 'tpu': self.device = self.__xm.xla_device()
+            from sys import setrecursionlimit
+            setrecursionlimit(self.epochs*self.batch_size)
             while feed_forward:
                 self.__model.train()
                 loss_item, total_train_loss = 1.0, 1.0
@@ -476,6 +480,18 @@ class SapiensGPT:
             except: pass
             try: return training_metrics
             except: return {'val_loss': 1.0, 'loss': 1.0, 'generalization_rate': 0.0, 'precision': 0.0}
+    def training(self, dataset_path='', progress=True):
+        try: return self.train(dataset_path=dataset_path, progress=progress)
+        except Exception as error:
+            try:
+                if self.__show_errors:
+                    error_message = 'ERROR in SapiensGPT.training: '+str(error)
+                    print(error_message)
+                    try: self.__print_exc() if self.__display_error_point else None
+                    except: pass
+            except: pass
+            try: return training_metrics
+            except: return {'val_loss': 1.0, 'loss': 1.0, 'generalization_rate': 0.0, 'precision': 0.0}        
     def saveModel(self, model_path='', progress=True):
         try:
             model_path = str(model_path).strip()

{sapiens_gpt-1.0.2 → sapiens_gpt-2.0.4}/sapiens_gpt.egg-info/PKG-INFO

@@ -1,16 +1,16 @@
 Metadata-Version: 2.2
 Name: sapiens_gpt
-Version: 1.0.2
+Version: 2.0.4
 Home-page: https://github.com/sapiens-technology/sapiens_gpt
 Author: SAPIENS TECHNOLOGY
 License: Proprietary Software
 License-File: LICENSE.txt
-Requires-Dist: torch==2.4.1
-Requires-Dist: torch-xla==2.7.0
-Requires-Dist: numpy==1.25.2
-Requires-Dist: tiktoken==0.4.0
+Requires-Dist: torch
+Requires-Dist: numpy
+Requires-Dist: tiktoken
 Requires-Dist: tqdm
 Requires-Dist: certifi
+Requires-Dist: torch-xla==2.7.0; platform_system == "Linux" and platform_machine == "x86_64"
 Dynamic: author
 Dynamic: home-page
 Dynamic: license

sapiens_gpt-2.0.4/sapiens_gpt.egg-info/requires.txt

@@ -0,0 +1,8 @@
+torch
+numpy
+tiktoken
+tqdm
+certifi
+
+[:platform_system == "Linux" and platform_machine == "x86_64"]
+torch-xla==2.7.0

{sapiens_gpt-1.0.2 → sapiens_gpt-2.0.4}/setup.py

@@ -5,13 +5,20 @@
 # --------------------------> A SAPIENS TECHNOLOGY®️ PRODUCTION) <--------------------------
 from setuptools import setup, find_packages
 package_name = 'sapiens_gpt'
-version = '1.0.2'
+version = '2.0.4'
 setup(
     name=package_name,
     version=version,
     author='SAPIENS TECHNOLOGY',
     packages=find_packages(),
-    install_requires=['torch==2.4.1', 'torch-xla==2.7.0', 'numpy==1.25.2', 'tiktoken==0.4.0', 'tqdm', 'certifi'],
+    install_requires=[
+        'torch',
+        'numpy',
+        'tiktoken',
+        'tqdm',
+        'certifi',
+        "torch-xla==2.7.0; platform_system=='Linux' and platform_machine=='x86_64'"
+    ],
     url='https://github.com/sapiens-technology/sapiens_gpt',
     license='Proprietary Software'
 )

sapiens_gpt-1.0.2/sapiens_gpt.egg-info/requires.txt

@@ -1,6 +0,0 @@
-torch==2.4.1
-torch-xla==2.7.0
-numpy==1.25.2
-tiktoken==0.4.0
-tqdm
-certifi