vsjetpack 2.2.0__tar.gz → 2.2.1__tar.gz

{vsjetpack-2.2.0 → vsjetpack-2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: vsjetpack
-Version: 2.2.0
+Version: 2.2.1
 Summary: Full suite of filters, wrappers, and helper functions for filtering video using VapourSynth
 Project-URL: Documentation, https://jaded-encoding-thaumaturgy.github.io/vs-jetpack/
 Project-URL: Source Code, https://github.com/Jaded-Encoding-Thaumaturgy/vs-jetpack
@@ -98,7 +98,7 @@ Requires-Dist: vapoursynth-fmtconv>=31; extra == 'deband'
 Requires-Dist: vapoursynth-hysteresis>=1.1.3; extra == 'deband'
 Requires-Dist: vapoursynth-lsmas>=1282.0.0; extra == 'deband'
 Requires-Dist: vapoursynth-mlrt-ov>=15.16; extra == 'deband'
-Requires-Dist: vapoursynth-mvutensils>=2; extra == 'deband'
+Requires-Dist: vapoursynth-mvutensils>=6; extra == 'deband'
 Requires-Dist: vapoursynth-nlm-ispc>=3; extra == 'deband'
 Requires-Dist: vapoursynth-resize2>=0.4.2; extra == 'deband'
 Requires-Dist: vapoursynth-sangnom>=43; extra == 'deband'
@@ -135,7 +135,7 @@ Requires-Dist: vapoursynth-fmtconv>=31; extra == 'dehalo'
 Requires-Dist: vapoursynth-hysteresis>=1.1.3; extra == 'dehalo'
 Requires-Dist: vapoursynth-lsmas>=1282.0.0; extra == 'dehalo'
 Requires-Dist: vapoursynth-mlrt-ov>=15.16; extra == 'dehalo'
-Requires-Dist: vapoursynth-mvutensils>=2; extra == 'dehalo'
+Requires-Dist: vapoursynth-mvutensils>=6; extra == 'dehalo'
 Requires-Dist: vapoursynth-nlm-ispc>=3; extra == 'dehalo'
 Requires-Dist: vapoursynth-resize2>=0.4.2; extra == 'dehalo'
 Requires-Dist: vapoursynth-sangnom>=43; extra == 'dehalo'
@@ -172,7 +172,7 @@ Requires-Dist: vapoursynth-fmtconv>=31; extra == 'deinterlace'
 Requires-Dist: vapoursynth-hysteresis>=1.1.3; extra == 'deinterlace'
 Requires-Dist: vapoursynth-lsmas>=1282.0.0; extra == 'deinterlace'
 Requires-Dist: vapoursynth-mlrt-ov>=15.16; extra == 'deinterlace'
-Requires-Dist: vapoursynth-mvutensils>=2; extra == 'deinterlace'
+Requires-Dist: vapoursynth-mvutensils>=6; extra == 'deinterlace'
 Requires-Dist: vapoursynth-nlm-ispc>=3; extra == 'deinterlace'
 Requires-Dist: vapoursynth-resize2>=0.4.2; extra == 'deinterlace'
 Requires-Dist: vapoursynth-sangnom>=43; extra == 'deinterlace'
@@ -209,7 +209,7 @@ Requires-Dist: vapoursynth-fmtconv>=31; extra == 'denoise'
 Requires-Dist: vapoursynth-hysteresis>=1.1.3; extra == 'denoise'
 Requires-Dist: vapoursynth-lsmas>=1282.0.0; extra == 'denoise'
 Requires-Dist: vapoursynth-mlrt-ov>=15.16; extra == 'denoise'
-Requires-Dist: vapoursynth-mvutensils>=2; extra == 'denoise'
+Requires-Dist: vapoursynth-mvutensils>=6; extra == 'denoise'
 Requires-Dist: vapoursynth-nlm-ispc>=3; extra == 'denoise'
 Requires-Dist: vapoursynth-resize2>=0.4.2; extra == 'denoise'
 Requires-Dist: vapoursynth-sangnom>=43; extra == 'denoise'
@@ -247,7 +247,7 @@ Requires-Dist: vapoursynth-fmtconv>=31; extra == 'full'
 Requires-Dist: vapoursynth-hysteresis>=1.1.3; extra == 'full'
 Requires-Dist: vapoursynth-lsmas>=1282.0.0; extra == 'full'
 Requires-Dist: vapoursynth-mlrt-ov>=15.16; extra == 'full'
-Requires-Dist: vapoursynth-mvutensils>=2; extra == 'full'
+Requires-Dist: vapoursynth-mvutensils>=6; extra == 'full'
 Requires-Dist: vapoursynth-nlm-ispc>=3; extra == 'full'
 Requires-Dist: vapoursynth-resize2>=0.4.2; extra == 'full'
 Requires-Dist: vapoursynth-sangnom>=43; extra == 'full'
@@ -282,18 +282,16 @@ Provides-Extra: nvidia
 Requires-Dist: cuda-python<14.0.0,>=13.0.0; extra == 'nvidia'
 Requires-Dist: dfttest2[cufft,nvrtc]>=10; extra == 'nvidia'
 Requires-Dist: ml-dtypes>=0.5.4; extra == 'nvidia'
-Requires-Dist: nvidia-cudnn-cu13; extra == 'nvidia'
 Requires-Dist: onnx>=1.21.0; extra == 'nvidia'
 Requires-Dist: onnxconverter-common>=1.16.0; extra == 'nvidia'
-Requires-Dist: tensorrt-cu13-libs; extra == 'nvidia'
 Requires-Dist: tensorrt-cu13>=11.0.0.114; extra == 'nvidia'
-Requires-Dist: tensorrt-rtx-cu13-bindings; extra == 'nvidia'
 Requires-Dist: tensorrt-rtx-cu13>=1.5.0.114; extra == 'nvidia'
 Requires-Dist: vapoursynth-bilateralgpu>=14; extra == 'nvidia'
 Requires-Dist: vapoursynth-bm3dcuda>=2.16; extra == 'nvidia'
 Requires-Dist: vapoursynth-eedi3vk2>=1.0; extra == 'nvidia'
 Requires-Dist: vapoursynth-mlrt-ncnn>=15.16; extra == 'nvidia'
-Requires-Dist: vapoursynth-mlrt-ort-cuda>=15.16.2; extra == 'nvidia'
+Requires-Dist: vapoursynth-mlrt-ort-cuda>=16.2; extra == 'nvidia'
+Requires-Dist: vapoursynth-mlrt-ort>=16.2; extra == 'nvidia'
 Requires-Dist: vapoursynth-mlrt-trt-rtx>=15.16.1; extra == 'nvidia'
 Requires-Dist: vapoursynth-mlrt-trt>=15.16.1; extra == 'nvidia'
 Requires-Dist: vapoursynth-nnedi3vk>=1.0; extra == 'nvidia'

{vsjetpack-2.2.0 → vsjetpack-2.2.1}/pyproject.toml

@@ -126,7 +126,7 @@ denoise = [
     "vapoursynth-fmtconv>=31",
     # TODO: uncomment when manipmv is bumped and add it to the README
     # "vapoursynth-manipmv>=1.6.0",
-    "vapoursynth-mvutensils>=2",
+    "vapoursynth-mvutensils>=6",
     "vapoursynth-nlm-ispc>=3",
 ]
 dehalo = ["vsjetpack[denoise]"]
@@ -153,16 +153,13 @@ nvidia = [
     "dfttest2[cufft,nvrtc]>=10",
     "vapoursynth-bm3dcuda>=2.16",
     "vapoursynth-bilateralgpu>=14",
-    "vapoursynth-mlrt-ort-cuda>=15.16.2",
+    "vapoursynth-mlrt-ort>=16.2",
+    "vapoursynth-mlrt-ort-cuda>=16.2",
     "vapoursynth-mlrt-trt>=15.16.1",
     "vapoursynth-mlrt-trt-rtx>=15.16.1",
     "vapoursynth-vszipcu>=1.2.0",
     "tensorrt-cu13>=11.0.0.114",
     "tensorrt-rtx-cu13>=1.5.0.114",
-    # Explicitly specify these so they're properly resolved through uv --index
-    "nvidia-cudnn-cu13",
-    "tensorrt-cu13-libs",
-    "tensorrt-rtx-cu13-bindings",
 ]
 amd = [
     "vsjetpack[gpu,cl,vulkan]",
@@ -221,6 +218,11 @@ doc = [
     # Additional packages to be documented.
     "vstransitions==0.1.4",
 ]
+index = [
+    # Explicitly specify these so they're properly resolved through uv --index
+    "nvidia-cudnn-cu13",
+    "tensorrt-cu13-libs",
+]
 
 [tool.vsscale.onnx.download]
 provider = ["ArtCNN==v1.6.2", "DPIR==20210902", "Waifu2x==20250502-2"]
@@ -231,9 +233,6 @@ vapoursynth-ffms2 = { index = "vs-wheels" }
 vapoursynth-fmtconv = { index = "vs-wheels" }
 nvidia-cudnn-cu13 = { index = "nvidia" }
 tensorrt-cu13-libs = { index = "nvidia" }
-tensorrt-rtx-cu13-bindings = [
-    { index = "vs-wheels", marker = "python_version == '3.14'" },
-]
 
 [[tool.uv.index]]
 # Not exactly necessary since nvidia wheels are also published on pypi as wheel stub (generic sdist)

{vsjetpack-2.2.0 → vsjetpack-2.2.1}/stubs/tensorrt/__init__.pyi

@@ -443,7 +443,7 @@ __all__: list[str] = [
     "volume",
     "warnings",
 ]
-__version__: str = "11.1.0.106"
+__version__: str = "11.2.1.2"
 attr: str = "VERBOSE"
 bfloat16: tensorrt_bindings.tensorrt.DataType  # value = <DataType.BF16: 7>
 bool: tensorrt_bindings.tensorrt.DataType  # value = <DataType.BOOL: 4>

{vsjetpack-2.2.0 → vsjetpack-2.2.1}/stubs/tensorrt_bindings/__init__.pyi

@@ -453,7 +453,6 @@ def _itemsize(trt_type):
     :arg trt_type: The TensorRT data type.
 
     :returns: The size of the type.
-
     """
 
 def common_enter(this): ...
@@ -462,7 +461,6 @@ def common_exit(this, exc_type, exc_value, traceback):
 
     Context managers are deprecated and have no effect. Objects are automatically freed when
     the reference count reaches 0.
-
     """
 
 def nptype(trt_type):
@@ -473,7 +471,6 @@ def nptype(trt_type):
     :arg trt_type: The TensorRT data type to convert.
 
     :returns: The equivalent numpy type.
-
     """
 
 def volume(iterable):
@@ -484,10 +481,9 @@ def volume(iterable):
     :arg iterable: Any python iterable, including a :class:`Dims` object.
 
     :returns: The volume of the iterable. This will return 1 for empty iterables, as a scalar has an empty shape and the volume of a tensor with empty shape is 1.
-
     """
 
-__version__: str = "11.1.0.106"
+__version__: str = "11.2.1.2"
 attr: str = "VERBOSE"
 bfloat16: tensorrt.DataType  # value = <DataType.BF16: 7>
 bool: tensorrt.DataType  # value = <DataType.BOOL: 4>

{vsjetpack-2.2.0 → vsjetpack-2.2.1}/stubs/tensorrt_bindings/tensorrt.pyi

@@ -1,9 +1,9 @@
 import builtins
 import collections.abc
+import types
 import typing
 
 import numpy
-import typing_extensions
 
 __all__: list[str] = [
     "APILanguage",
@@ -466,7 +466,6 @@ class Builder:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __del__(self) -> None: ...
     def __init__(self, logger: ILogger) -> None:
@@ -621,7 +620,7 @@ class BuilderFlag:
 
       DISABLE_COMPILATION_CACHE : Disable caching JIT compilation results during engine build.
 
-      STRIP_PLAN : Strip the refittable weights from the engine plan file.
+      STRIP_PLAN : Strip refittable weights from the engine plan file. If no refit mode is specified, REFIT_IDENTICAL is enabled by default. When used with REFIT_INDIVIDUAL, only weights explicitly marked with INetworkDefinition.mark_weights_refittable() are stripped. When used with REFIT or REFIT_IDENTICAL, TensorRT determines which refittable weights are stripped according to the selected refit mode.
 
       REFIT_IDENTICAL : Create a refittable engine using identical weights. Different weights during refits yield unpredictable behavior.
 
@@ -1745,7 +1744,7 @@ class IBuilderConfig:
     :ivar tiling_optimization_level: The optimization level of tiling strategies. A Higher level allows TensorRT to spend more time searching for better optimization strategy.
     :ivar l2_limit_for_tiling: The target L2 cache usage for tiling optimization.
     :ivar remote_auto_tuning_config: The config string to be used during remote auto-tuning. Remote auto-tuning is only enabled for engines built with EngineCapability.SAFETY.
-    :ivar build_route: :class:`str` The build route string passed to the compiler. The build route is a whitespace-separated list of ``-name=value`` Myelin knob tokens (e.g. ``"-conv_use_long_w=off -kgen:codegen:cuda_tile=2"``) used to customize engine compilation for performance tuning. Knob names are validated against the list returned by :attr:`all_build_routes` unless the internal ``TRT_BYPASS_BUILD_ROUTE_WHITELIST`` flag is set. Setting an empty string resets the build route. Available only when the global performance tuner feature is enabled in the build (disabled by default on RTX/Windows targets).
+    :ivar build_route: :class:`str` The build route string passed to the compiler. The build route is a whitespace-separated list of ``-name=value`` Myelin knob tokens (e.g. ``"-conv_use_long_w=off -kgen:codegen:cuda_tile=2"``) used to customize engine compilation for performance tuning. Knob names are validated against the list returned by :attr:`all_build_routes` unless the internal ``bypass_build_route_whitelist`` option (set via ``TRT_INTERNAL_OPTIONS``) is enabled. Setting an empty string resets the build route. Available only when the global performance tuner feature is enabled in the build (disabled by default on RTX/Windows targets).
     :ivar all_build_routes: :class:`str` JSON description of every build-route knob the compiler supports, populated when the :class:`IBuilderConfig` is created. Read-only. Returns an empty string on platforms where the global performance tuner feature is disabled (e.g. RTX/Windows targets).
 
     Below are the descriptions about each builder optimization level:
@@ -1776,7 +1775,6 @@ class IBuilderConfig:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __del__(self) -> None: ...
     def add_optimization_profile(self, profile: IOptimizationProfile) -> int:
@@ -2119,6 +2117,7 @@ class ICudaEngine:
     :ivar streamable_weights_size: Returns the size of the streamable weights in the engine. This may not include all the weights.
     :ivar weight_streaming_budget_v2: Set and get the current weight streaming budget for inference. The budget may be set any non-negative value. A value of 0 streams the most weights. Values equal to streamable_weights_size (default) or larger will disable weight streaming.
     :ivar weight_streaming_scratch_memory_size: The amount of scratch memory required by a TensorRT ExecutionContext to perform inference. This value may change based on the current weight streaming budget. Please use the V2 memory APIs, engine.device_memory_size_v2 and ExecutionContext.set_device_memory() to provide memory which includes the current weight streaming scratch memory. Not specifying these APIs or using the V1 APIs will not include this memory, so TensorRT will resort to allocating itself.
+    :ivar weights_loaded: Whether GPU weights are currently loaded and ready for inference. False after deferred deserialization, True after load_weights() succeeds.
 
     """
 
@@ -2131,7 +2130,6 @@ class ICudaEngine:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __del__(self) -> None: ...
     def __getitem__(self, arg0: typing.SupportsInt) -> str: ...
@@ -2443,7 +2441,7 @@ class IDebugListener:
     def __init__(self) -> None: ...
     def process_debug_tensor(
         self,
-        addr: typing_extensions.CapsuleType,
+        addr: types.CapsuleType,
         location: TensorLocation,
         type: DataType,
         shape: Dims,
@@ -2785,7 +2783,6 @@ class IExecutionContext:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __del__(self) -> None: ...
     def execute_async_v3(self, stream_handle: typing.SupportsInt) -> builtins.bool:
@@ -2912,7 +2909,7 @@ class IExecutionContext:
 
         :arg aux_streams: A list of cuda streams. If the length of the list is greater than engine.num_aux_streams, then only the first "engine.num_aux_streams" streams will be used. If the length is less than engine.num_aux_streams, such as an empty list, then TensorRT will use the provided streams for the first few auxiliary streams, and will create additional streams internally for the rest of the auxiliary streams.
         """
-    def set_communicator(self, communicator: typing_extensions.CapsuleType) -> builtins.bool:
+    def set_communicator(self, communicator: types.CapsuleType) -> builtins.bool:
         """
         Set the NCCL communicator for the execution context.
 
@@ -3086,7 +3083,6 @@ class IFillLayer(ILayer):
     alpha: typing.Any
     beta: typing.Any
     operation: FillOperation
-    shape: Dims
     to_type: DataType
     def is_alpha_beta_int64(self) -> builtins.bool: ...
     def set_input(self, index: typing.SupportsInt, tensor: ITensor) -> None:
@@ -3120,6 +3116,10 @@ class IFillLayer(ILayer):
         :arg index: the index of the input to modify.
         :arg tensor: the input tensor.
         """
+    @property
+    def shape(self) -> typing.Any: ...
+    @shape.setter
+    def shape(self, arg1: Dims) -> None: ...
 
 class IGatherLayer(ILayer):
     """
@@ -3159,7 +3159,7 @@ class IGpuAllocator(IVersionedInterface):
     def __init__(self) -> None: ...
     def allocate(
         self, size: typing.SupportsInt, alignment: typing.SupportsInt, flags: typing.SupportsInt
-    ) -> typing_extensions.CapsuleType:
+    ) -> types.CapsuleType:
         """
         [DEPRECATED] Deprecated in TensorRT 10.0. Please use allocate_async instead.
         A callback implemented by the application to handle acquisition of GPU memory.
@@ -3182,7 +3182,7 @@ class IGpuAllocator(IVersionedInterface):
         alignment: typing.SupportsInt,
         flags: typing.SupportsInt,
         stream: typing.SupportsInt,
-    ) -> typing_extensions.CapsuleType:
+    ) -> types.CapsuleType:
         """
         A callback implemented by the application to handle acquisition of GPU memory asynchronously.
         This is just a wrapper around a syncronous method allocate.
@@ -3201,7 +3201,7 @@ class IGpuAllocator(IVersionedInterface):
 
         :returns: The address of the allocated memory
         """
-    def deallocate(self, memory: typing_extensions.CapsuleType) -> builtins.bool:
+    def deallocate(self, memory: types.CapsuleType) -> builtins.bool:
         """
         [DEPRECATED] Deprecated in TensorRT 10.0. Please use dealocate_async instead;
         A callback implemented by the application to handle release of GPU memory.
@@ -3212,7 +3212,7 @@ class IGpuAllocator(IVersionedInterface):
 
         :returns: True if the acquired memory is released successfully.
         """
-    def deallocate_async(self, memory: typing_extensions.CapsuleType, stream: typing.SupportsInt) -> builtins.bool:
+    def deallocate_async(self, memory: types.CapsuleType, stream: typing.SupportsInt) -> builtins.bool:
         """
         A callback implemented by the application to handle release of GPU memory asynchronously.
         This is just a wrapper around a syncronous method deallocate.
@@ -3226,8 +3226,8 @@ class IGpuAllocator(IVersionedInterface):
         :returns: True if the acquired memory is released successfully.
         """
     def reallocate(
-        self, address: typing_extensions.CapsuleType, alignment: typing.SupportsInt, new_size: typing.SupportsInt
-    ) -> typing_extensions.CapsuleType:
+        self, address: types.CapsuleType, alignment: typing.SupportsInt, new_size: typing.SupportsInt
+    ) -> types.CapsuleType:
         """
         A callback implemented by the application to resize an existing allocation.
 
@@ -3274,7 +3274,7 @@ class IGpuAsyncAllocator(IGpuAllocator):
     def __init__(self) -> None: ...
     def allocate(
         self, size: typing.SupportsInt, alignment: typing.SupportsInt, flags: typing.SupportsInt
-    ) -> typing_extensions.CapsuleType:
+    ) -> types.CapsuleType:
         """
         [DEPRECATED] Deprecated in TensorRT 10.0. Please use allocate_async instead.
         A callback implemented by the application to handle acquisition of GPU memory.
@@ -3299,7 +3299,7 @@ class IGpuAsyncAllocator(IGpuAllocator):
         alignment: typing.SupportsInt,
         flags: typing.SupportsInt,
         stream: typing.SupportsInt,
-    ) -> typing_extensions.CapsuleType:
+    ) -> types.CapsuleType:
         """
         A callback implemented by the application to handle acquisition of GPU memory asynchronously.
         If an allocation request of size 0 is made, ``None`` should be returned.
@@ -3316,7 +3316,7 @@ class IGpuAsyncAllocator(IGpuAllocator):
 
         :returns: The address of the allocated memory
         """
-    def deallocate(self, memory: typing_extensions.CapsuleType) -> builtins.bool:
+    def deallocate(self, memory: types.CapsuleType) -> builtins.bool:
         """
         [DEPRECATED] Deprecated in TensorRT 10.0. Please use deallocate_async instead.
         A callback implemented by the application to handle release of GPU memory.
@@ -3328,9 +3328,7 @@ class IGpuAsyncAllocator(IGpuAllocator):
 
         :returns: True if the acquired memory is released successfully.
         """
-    def deallocate_async(
-        self: IGpuAllocator, memory: typing_extensions.CapsuleType, stream: typing.SupportsInt
-    ) -> builtins.bool:
+    def deallocate_async(self: IGpuAllocator, memory: types.CapsuleType, stream: typing.SupportsInt) -> builtins.bool:
         """
         A callback implemented by the application to handle release of GPU memory asynchronously.
 
@@ -3348,7 +3346,7 @@ class IGridSampleLayer(ILayer):
     A grid sample layer in an :class:`INetworkDefinition` .
 
     This layer uses an input tensor and a grid tensor to produce an interpolated output tensor.
-    The input and grid tensors must shape tensors of rank 4. The only supported `SampleMode` s are
+    The input and grid tensors must be tensors of rank 4 or 5. The only supported `SampleMode` s are
     trt.samplemode.CLAMP, trt.samplemode.FILL, and trt.samplemode.REFLECT.
 
     :ivar interpolation_mode: class:`InterpolationMode` The interpolation type to use. Defaults to LINEAR.
@@ -3380,7 +3378,6 @@ class IHostMemory:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __buffer__(self, flags):
         """
@@ -4067,7 +4064,6 @@ class INetworkDefinition:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __del__(self) -> None: ...
     def __getitem__(self, arg0: typing.SupportsInt) -> ILayer: ...
@@ -4344,7 +4340,7 @@ class INetworkDefinition:
         """
     def add_grid_sample(self, input: ITensor, grid: ITensor) -> IGridSampleLayer:
         """
-        Creates a GridSample layer with a trt.InterpolationMode.LINEAR, unaligned corners, and trt.SampleMode.FILL for 4d-shape input tensors.
+        Creates a GridSample layer with a trt.InterpolationMode.LINEAR, unaligned corners, and trt.SampleMode.FILL for rank 4 or 5 input tensors.
         See :class:`IGridSampleLayer` for more information.
 
         :arg input: The input tensor to the layer.
@@ -5461,7 +5457,7 @@ class IPluginRegistry:
         :returns: ``True`` if the plugin creator was deregistered, ``False`` if it was not found in the registry
                 or otherwise could not be deregistered.
         """
-    def deregister_library(self, handle: typing_extensions.CapsuleType) -> None:
+    def deregister_library(self, handle: types.CapsuleType) -> None:
         """
         Deregister plugins associated with a library. Any resources acquired when the library was loaded will be released.
 
@@ -5477,7 +5473,7 @@ class IPluginRegistry:
 
         :returns: An :class:`IPluginCreator` .
         """
-    def load_library(self, plugin_path: str) -> typing_extensions.CapsuleType:
+    def load_library(self, plugin_path: str) -> types.CapsuleType:
         """
         Load and register a shared library of plugins.
 
@@ -5601,9 +5597,9 @@ class IPluginV2:
     def execute_async(
         self,
         batch_size: typing.SupportsInt,
-        inputs: collections.abc.Sequence[typing_extensions.CapsuleType],
-        outputs: collections.abc.Sequence[typing_extensions.CapsuleType],
-        workspace: typing_extensions.CapsuleType,
+        inputs: collections.abc.Sequence[types.CapsuleType],
+        outputs: collections.abc.Sequence[types.CapsuleType],
+        workspace: types.CapsuleType,
         stream_handle: typing.SupportsInt,
     ) -> int:
         """
@@ -5888,10 +5884,7 @@ class IPluginV2Ext(IPluginV2):
     :ivar tensorrt_version: :class:`int` The API version with which this plugin was built.
     """
     def attach_to_context(
-        self,
-        cudnn: typing_extensions.CapsuleType,
-        cublas: typing_extensions.CapsuleType,
-        allocator: typing_extensions.CapsuleType,
+        self, cudnn: types.CapsuleType, cublas: types.CapsuleType, allocator: types.CapsuleType
     ) -> None:
         """
         Attach the plugin object to an execution context and grant the plugin the access to some context resource.
@@ -6578,7 +6571,7 @@ class IResizeLayer(ILayer):
        * (ResizeCoordinateTransformation.HALF_PIXEL, ResizeSelector.UPPER)
 
 
-    :ivar shape: :class:`Dims` The output dimensions. Must to equal to input dimensions size.
+    :ivar shape: :class:`Dims` The output dimensions, or ``None`` if they are specified dynamically through a second layer input or via scales instead. Must be equal to the input dimensions size.
     :ivar scales: :class:`List[float]` List of resize scales.
         If executing this layer on DLA, there are three restrictions:
         1. ``len(scales)`` has to be exactly 4.
@@ -6599,7 +6592,6 @@ class IResizeLayer(ILayer):
     nearest_rounding: ResizeRoundMode
     resize_mode: InterpolationMode
     selector_for_single_pixel: ResizeSelector
-    shape: Dims
     def set_input(self, index: typing.SupportsInt, tensor: ITensor) -> None:
         """
         Sets the input tensor for the given index.
@@ -6620,6 +6612,10 @@ class IResizeLayer(ILayer):
     def scales(self) -> list[float]: ...
     @scales.setter
     def scales(self, arg1: collections.abc.Sequence[typing.SupportsFloat]) -> None: ...
+    @property
+    def shape(self) -> typing.Any: ...
+    @shape.setter
+    def shape(self, arg1: Dims) -> None: ...
 
 class IReverseSequenceLayer(ILayer):
     """
@@ -6796,7 +6792,7 @@ class IShuffleLayer(ILayer):
     This class shuffles data by applying in sequence: a transpose operation, a reshape operation and a second transpose operation. The dimension types of the output are those of the reshape dimension.
 
     :ivar first_transpose: :class:`Permutation` The permutation applied by the first transpose operation. Default: Identity Permutation
-    :ivar reshape_dims: :class:`Dims` The reshaped dimensions.
+    :ivar reshape_dims: :class:`Dims` The reshaped dimensions, or ``None`` if they are specified dynamically through a second layer input instead.
         Two special values can be used as dimensions.
         Value 0 copies the corresponding dimension from input. This special value can be used more than once in the dimensions. If number of reshape dimensions is less than input, 0s are resolved by aligning the most significant dimensions of input.
         Value -1 infers that particular dimension by looking at input and rest of the reshape dimensions. Note that only a maximum of one dimension is permitted to be specified as -1.
@@ -6809,7 +6805,6 @@ class IShuffleLayer(ILayer):
     """
 
     first_transpose: Permutation
-    reshape_dims: Dims
     second_transpose: Permutation
     zero_is_placeholder: builtins.bool
     def set_input(self, index: typing.SupportsInt, tensor: ITensor) -> None:
@@ -6834,6 +6829,10 @@ class IShuffleLayer(ILayer):
         :arg index: The index of the input tensor.
         :arg tensor: The input tensor.
         """
+    @property
+    def reshape_dims(self) -> typing.Any: ...
+    @reshape_dims.setter
+    def reshape_dims(self, arg1: Dims) -> None: ...
 
 class ISliceLayer(ILayer):
     """
@@ -6889,18 +6888,14 @@ class ISliceLayer(ILayer):
     * The input tensor has four dimensions.
     * For :const:`SliceMode.FILL` , the fill value input is a scalar output of an :class:`IConstantLayer` with value 0 that is not consumed by any other layer.
 
-    :ivar start: :class:`Dims` The start offset.
-    :ivar shape: :class:`Dims` The output dimensions.
-    :ivar stride: :class:`Dims` The slicing stride.
+    :ivar start: :class:`Dims` The start offset, or ``None`` if it is specified dynamically through a layer input instead.
+    :ivar shape: :class:`Dims` The output dimensions, or ``None`` if they are specified dynamically through a layer input instead.
+    :ivar stride: :class:`Dims` The slicing stride, or ``None`` if it is specified dynamically through a layer input instead.
     :ivar mode: :class:`SampleMode` Controls how :class:`ISliceLayer` handles out of bounds coordinates.
-    :ivar axes: :class:`Dims` The axes that starts, sizes, and strides correspond to.
+    :ivar axes: :class:`Dims` The axes that starts, sizes, and strides correspond to, or ``None`` if they are specified dynamically through a layer input instead.
     """
 
-    axes: Dims
     mode: SampleMode
-    shape: Dims
-    start: Dims
-    stride: Dims
     def set_input(self, index: typing.SupportsInt, tensor: ITensor) -> None:
         """
         Sets the input tensor for the given index. The index must be 0 or 4 for a static slice layer.
@@ -6926,6 +6921,22 @@ class ISliceLayer(ILayer):
         :arg index: The index of the input tensor.
         :arg tensor: The input tensor.
         """
+    @property
+    def axes(self) -> typing.Any: ...
+    @axes.setter
+    def axes(self, arg1: Dims) -> None: ...
+    @property
+    def shape(self) -> typing.Any: ...
+    @shape.setter
+    def shape(self, arg1: Dims) -> None: ...
+    @property
+    def start(self) -> typing.Any: ...
+    @start.setter
+    def start(self, arg1: Dims) -> None: ...
+    @property
+    def stride(self) -> typing.Any: ...
+    @stride.setter
+    def stride(self, arg1: Dims) -> None: ...
 
 class ISoftMaxLayer(ILayer):
     """
@@ -7027,9 +7038,7 @@ class IStreamReaderV2:
             return data
     """
     def __init__(self) -> None: ...
-    def read(
-        self, destination: typing_extensions.CapsuleType, num_bytes: typing.SupportsInt, stream: typing.SupportsInt
-    ) -> int:
+    def read(self, destination: types.CapsuleType, num_bytes: typing.SupportsInt, stream: typing.SupportsInt) -> int:
         """
         A callback implemented by the application to set the stream location.
 
@@ -7064,7 +7073,7 @@ class IStreamWriter:
 
     """
     def __init__(self) -> None: ...
-    def write(self, data: typing_extensions.CapsuleType, size: typing.SupportsInt) -> int:
+    def write(self, data: types.CapsuleType, size: typing.SupportsInt) -> int:
         """
         A callback implemented by the application to write a particular chunk of memory.
 
@@ -8021,7 +8030,6 @@ class OnnxParser:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __del__(self) -> None: ...
     def __init__(self, network: INetworkDefinition, logger: ILogger) -> None:
@@ -9283,6 +9291,7 @@ class Runtime:
     :ivar tempfile_control_flags: :class:`int` Flags which control whether TensorRT is allowed to create in-memory or temporary files.
                                                See :class:`TempfileControlFlag` for details.
     :ivar engine_host_code_allowed: :class:`builtins.bool` Whether this runtime is allowed to deserialize engines that contain host executable code (Default: False).
+    :ivar defer_weights_loading: :class:`builtins.bool` When True, the next deserialize_cuda_engine() call will skip GPU weight allocation. Call ICudaEngine.load_weights() or ICudaEngine.load_weights_async() to bring weights to the GPU when ready (Default: False).
 
     """
 
@@ -9297,7 +9306,6 @@ class Runtime:
 
         Context managers are deprecated and have no effect. Objects are automatically freed when
         the reference count reaches 0.
-
         """
     def __del__(self) -> None: ...
     def __init__(self, logger: ILogger) -> None:
@@ -10227,7 +10235,7 @@ def get_plugin_registry() -> IPluginRegistry:
     Return the plugin registry for standard runtime
     """
 
-def init_libnvinfer_plugins(logger: typing_extensions.CapsuleType, namespace: str) -> builtins.bool:
+def init_libnvinfer_plugins(logger: types.CapsuleType, namespace: str) -> builtins.bool:
     """
     Initialize and register all the existing TensorRT plugins to the :class:`IPluginRegistry` with an optional namespace.
     The plugin library author should ensure that this function name is unique to the library.

{vsjetpack-2.2.0 → vsjetpack-2.2.1}/stubs/tensorrt_rtx/__init__.pyi

@@ -445,7 +445,7 @@ __all__: list[str] = [
     "volume",
     "warnings",
 ]
-__version__: str = "1.5.0.114"
+__version__: str = "1.6.1.120"
 attr: str = "VERBOSE"
 bfloat16: tensorrt_rtx_bindings.tensorrt_rtx.DataType  # value = <DataType.BF16: 7>
 bool: tensorrt_rtx_bindings.tensorrt_rtx.DataType  # value = <DataType.BOOL: 4>

{vsjetpack-2.2.0 → vsjetpack-2.2.1}/stubs/tensorrt_rtx_bindings/__init__.pyi

@@ -455,7 +455,6 @@ def _itemsize(trt_type):
     :arg trt_type: The TensorRT data type.
 
     :returns: The size of the type.
-
     """
 
 def common_enter(this): ...
@@ -464,7 +463,6 @@ def common_exit(this, exc_type, exc_value, traceback):
 
     Context managers are deprecated and have no effect. Objects are automatically freed when
     the reference count reaches 0.
-
     """
 
 def nptype(trt_type):
@@ -475,7 +473,6 @@ def nptype(trt_type):
     :arg trt_type: The TensorRT data type to convert.
 
     :returns: The equivalent numpy type.
-
     """
 
 def volume(iterable):
@@ -486,10 +483,9 @@ def volume(iterable):
     :arg iterable: Any python iterable, including a :class:`Dims` object.
 
     :returns: The volume of the iterable. This will return 1 for empty iterables, as a scalar has an empty shape and the volume of a tensor with empty shape is 1.
-
     """
 
-__version__: str = "1.5.0.114"
+__version__: str = "1.6.1.120"
 attr: str = "VERBOSE"
 bfloat16: tensorrt_rtx.DataType  # value = <DataType.BF16: 7>
 bool: tensorrt_rtx.DataType  # value = <DataType.BOOL: 4>