pms-nvidia-processor 1.5.2__tar.gz → 2.0.0__tar.gz

{pms_nvidia_processor-1.5.2 → pms_nvidia_processor-2.0.0}/PKG-INFO

@@ -1,20 +1,17 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: pms-nvidia-processor
-Version: 1.5.2
+Version: 2.0.0
 Summary: lib for pms nvidia processors
 Author: HyeongSeok Kim
 Author-email: tiryul@gmail.com
-Requires-Python: >=3.10,<4.0
+Requires-Python: ==3.12.*
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: albumentations (>=1.4.13,<2.0.0)
 Requires-Dist: loguru
 Requires-Dist: numpy
-Requires-Dist: pms-inference-engine (>=1.2.0,<1.3.0)
-Requires-Dist: pms-tensorrt (>=1.3.0,<1.4.0)
+Requires-Dist: pms-inference-engine (>=2.0.0,<2.1.0)
+Requires-Dist: pms-tensorrt (>=2.0.0,<2.1.0)
 Description-Content-Type: text/markdown
 
 # pms-nvidia-processor

{pms_nvidia_processor-1.5.2 → pms_nvidia_processor-2.0.0}/pms_nvidia_processor/__init__.py

@@ -8,4 +8,4 @@ from ._color_resnet_pre_processor import COLORRESNETPREProcessor
 from ._color_resnet_post_processor import COLORRESNETPOSTProcessor
 from ._gg_processor import GGProcessor
 
-__version__ = "1.5.2"
+__version__ = "2.0.0"

{pms_nvidia_processor-1.5.2 → pms_nvidia_processor-2.0.0}/pms_nvidia_processor/_const.py

@@ -244,10 +244,9 @@ class ColorResnetPostConfig:
     NUMBER_OF_OUTPUT_CHANNELS: int = 3
     UPSCALE_RATIO: int = 1
     PATCH_SIZE = 512
-    # MAX_BATCH_SIZE = 8
     MAX_BATCH_SIZE = 1
     MIN_BATCH_SIZE = 1
-    OPT_BATCH_SIZE = MAX_BATCH_SIZE // 2
+    OPT_BATCH_SIZE = 1
     INPUT_OVERLAB_LENGTH = 16 * 2
 
     PATCHER_CONFIG = PatcherIOConfig(

{pms_nvidia_processor-1.5.2 → pms_nvidia_processor-2.0.0}/pyproject.toml

@@ -1,14 +1,14 @@
 [tool.poetry]
 name = "pms-nvidia-processor"
-version = "1.5.2"
+version = "2.0.0"
 description = "lib for pms nvidia processors"
 authors = ["HyeongSeok Kim <tiryul@gmail.com>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = "^3.10"
-pms-inference-engine = ">=1.2.0, <1.3.0"
-pms-tensorrt = ">=1.3.0, <1.4.0"
+python = "3.12.*"
+pms-inference-engine = ">=2.0.0, <2.1.0"
+pms-tensorrt = ">=2.0.0, <2.1.0"
 loguru = "*"
 numpy = "*"
 albumentations = "^1.4.13"
@@ -18,16 +18,12 @@ albumentations = "^1.4.13"
 pytest = "^8.1.1"
 pytest-order = "^1.2.0"
 pms-model-manager = "^1.3.1"
-tensorrt = "10.2.0.post1"
+tensorrt = "10.7.0"
 pycuda = "2024.1"
-tensorrt-cu12-libs = "10.2.0.post1"
-tensorrt-cu12-bindings = "10.2.0.post1"
+tensorrt-cu12-libs = "10.7.0"
+tensorrt-cu12-bindings = "10.7.0"
 
 
-[[tool.poetry.source]]
-name = "nvidia"
-url = "https://pypi.nvidia.com/"
-priority = "supplemental"
 
 [build-system]
 requires = ["poetry-core"]