pycoze 0.1.89__tar.gz → 0.1.90__tar.gz

{pycoze-0.1.89 → pycoze-0.1.90}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.89
+Version: 0.1.90
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

pycoze-0.1.90/pycoze/ai/__init__.py

@@ -0,0 +1 @@
+from .vram_reserve import reserve_vram, reserve_vram_retry, unreserve_vram

{pycoze-0.1.89 → pycoze-0.1.90}/pycoze/ai/vram_reserve.py

@@ -88,33 +88,25 @@ def reserve_vram_retry(gb, retry=None, uid=None):
         retry = sys.maxsize
     for i in range(retry):
         time.sleep(1)
-        if i % 10 == 0 or i < 10:
+        if i % 10 == 0 or i < 10 and i != 0:
             print(f"重试第{i}次")
         if reserve_vram(gb, uid):
             return True
     return False
 
 
-def release_vram(uid=None):
+def unreserve_vram(uid=None):
     if uid is None:
         uid = f"pid:{os.getpid()}"
     with sqlite3.connect(DATABASE_PATH) as conn:
         cursor = conn.cursor()
         cursor.execute(f"DELETE FROM {TABLE_NAME} WHERE uid = ?", (uid,))
         conn.commit()
-        # 计算释放后的剩余VRAM大小
-        cursor.execute(f"SELECT SUM(reserved_gb) FROM {TABLE_NAME}")
-        total_reserved = cursor.fetchone()[0]
-        if total_reserved is None:
-            total_reserved = 0.0
-        available_gb = get_vram_resources() - total_reserved
-        print(f"释放成功，剩余VRAM大小: {available_gb} GB")
 
 
 # 注册退出时的清理函数
 def cleanup():
-    release_vram()
-    print("程序退出，VRAM资源已释放")
+    unreserve_vram()
 
 
 def initialize_and_check():

{pycoze-0.1.89 → pycoze-0.1.90}/pycoze.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.89
+Version: 0.1.90
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.89 → pycoze-0.1.90}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pycoze",
-    version="0.1.89",
+    version="0.1.90",
     packages=find_packages(),
     install_requires=[],
     author="Yuan Jie Xiong",

pycoze-0.1.89/pycoze/ai/__init__.py

@@ -1 +0,0 @@
-from .vram_reserve import reserve_vram_retry, release_vram, reserve_vram