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

gptmodel/gptmodel.py

@@ -372,7 +372,9 @@ class GPTModel:
             else: directory, file_name = str(model_path), 'model.gpt'
             model_file = self.__os_path.join(directory, file_name)
             if progress:
-                for _ in self.__tqdm(range(10), desc='Loading model', leave=False): checkpoint = self.__load(model_file, map_location=self.__device)
+                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)
             try: self.__tokenizer = str(checkpoint['tokenizer']).lower().strip()
             except: self.__tokenizer = 'gpt'
@@ -445,7 +447,7 @@ class GPTModel:
             max_tokens = max((1, int(max_tokens))) if type(max_tokens) in (bool, int, float) else 500
             temperature = max((0, float(temperature))) if type(temperature) in (bool, int, float) else 0.5
             top_k = max((0, int(top_k))) if type(top_k) in (bool, int, float) else 0
-            top_p = min((1.0, max((0.0, int(top_p))))) if type(top_p) in (bool, int, float) else 1.0
+            top_p = min((1.0, max((0.0, float(top_p))))) if type(top_p) in (bool, int, float) else 1.0
             stream = bool(stream) if type(stream) in (bool, int, float) else False
             if self.__model is None: raise ValueError('Model is not initialized. Call train or loadModel first.')
             if stream: return self.__generate_tokens(prompt=prompt, max_tokens=max_tokens, temperature=temperature, top_k=top_k, top_p=top_p)
@@ -460,7 +462,7 @@ class GPTModel:
             max_tokens = max((1, int(max_tokens))) if type(max_tokens) in (bool, int, float) else 500
             temperature = max((0, float(temperature))) if type(temperature) in (bool, int, float) else 0.5
             top_k = max((0, int(top_k))) if type(top_k) in (bool, int, float) else 0
-            top_p = min((1.0, max((0.0, int(top_p))))) if type(top_p) in (bool, int, float) else 1.0
+            top_p = min((1.0, max((0.0, float(top_p))))) if type(top_p) in (bool, int, float) else 1.0
             stream = bool(stream) if type(stream) in (bool, int, float) else False
             if self.__model is None: raise ValueError('Model is not initialized. Call train or loadModel first.')
             if stream:

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

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

gptmodel-1.0.6.dist-info/RECORD

@@ -0,0 +1,7 @@
+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,,

gptmodel-1.0.4.dist-info/RECORD

@@ -1,7 +0,0 @@
-gptmodel/__init__.py,sha256=ifnJnBPHcBZwiHgt_mNp9gd1OJYCCHB3bgml6TmKBIw,1358
-gptmodel/gptmodel.py,sha256=3qbXbCEkXSsakFAuLu6RCacyHhdu9nIHagNriEi9NEE,33414
-gptmodel-1.0.4.dist-info/LICENSE.txt,sha256=WqB2vIA5tH5lqLTr53yT_oy1m0wYfuvCPQKxdDHWimg,115
-gptmodel-1.0.4.dist-info/METADATA,sha256=23KkOothlS852QRWKN6xP7BrKj1cLh0SQ723fet3Mps,384
-gptmodel-1.0.4.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-gptmodel-1.0.4.dist-info/top_level.txt,sha256=585C0QclguIkVPKKPpoeD2FxYAc5n5EAuvJNK4vMeQk,9
-gptmodel-1.0.4.dist-info/RECORD,,