paddlex 3.0.0rc0__py3-none-any.whl → 3.0.0rc1__py3-none-any.whl

{paddlex-3.0.0rc0.dist-info → paddlex-3.0.0rc1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: setuptools (79.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

paddlex/inference/models/base/predictor/basic_predictor.py

@@ -1,139 +0,0 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from typing import Dict, Any, Iterator
-from abc import abstractmethod
-
-from .....utils.subclass_register import AutoRegisterABCMetaClass
-from .....utils.flags import (
-    INFER_BENCHMARK,
-    INFER_BENCHMARK_WARMUP,
-)
-from .....utils import logging
-from ....utils.pp_option import PaddlePredictorOption
-from ....utils.benchmark import benchmark
-from .base_predictor import BasePredictor
-
-
-class BasicPredictor(
-    BasePredictor,
-    metaclass=AutoRegisterABCMetaClass,
-):
-    """BasicPredictor."""
-
-    __is_base = True
-
-    def __init__(
-        self,
-        model_dir: str,
-        config: Dict[str, Any] = None,
-        device: str = None,
-        batch_size: int = 1,
-        pp_option: PaddlePredictorOption = None,
-    ) -> None:
-        """Initializes the BasicPredictor.
-
-        Args:
-            model_dir (str): The directory where the model files are stored.
-            config (Dict[str, Any], optional): The configuration dictionary. Defaults to None.
-            device (str, optional): The device to run the inference engine on. Defaults to None.
-            batch_size (int, optional): The batch size to predict. Defaults to 1.
-            pp_option (PaddlePredictorOption, optional): The inference engine options. Defaults to None.
-        """
-        super().__init__(model_dir=model_dir, config=config)
-        if not pp_option:
-            pp_option = PaddlePredictorOption(model_name=self.model_name)
-        if device:
-            pp_option.device = device
-        trt_dynamic_shapes = (
-            self.config.get("Hpi", {})
-            .get("backend_configs", {})
-            .get("paddle_infer", {})
-            .get("trt_dynamic_shapes", None)
-        )
-        if trt_dynamic_shapes:
-            pp_option.trt_dynamic_shapes = trt_dynamic_shapes
-        self.pp_option = pp_option
-        self.pp_option.batch_size = batch_size
-        self.batch_sampler.batch_size = batch_size
-
-        logging.debug(f"{self.__class__.__name__}: {self.model_dir}")
-        self.benchmark = benchmark
-
-    def __call__(
-        self,
-        input: Any,
-        batch_size: int = None,
-        device: str = None,
-        pp_option: PaddlePredictorOption = None,
-        **kwargs: Dict[str, Any],
-    ) -> Iterator[Any]:
-        """
-        Predict with the input data.
-
-        Args:
-            input (Any): The input data to be predicted.
-            batch_size (int, optional): The batch size to use. Defaults to None.
-            device (str, optional): The device to run the predictor on. Defaults to None.
-            pp_option (PaddlePredictorOption, optional): The predictor options to set. Defaults to None.
-            **kwargs (Dict[str, Any]): Additional keyword arguments to set up predictor.
-
-        Returns:
-            Iterator[Any]: An iterator yielding the prediction output.
-        """
-        self.set_predictor(batch_size, device, pp_option)
-        if self.benchmark:
-            self.benchmark.start()
-            if INFER_BENCHMARK_WARMUP > 0:
-                output = self.apply(input, **kwargs)
-                warmup_num = 0
-                for _ in range(INFER_BENCHMARK_WARMUP):
-                    try:
-                        next(output)
-                        warmup_num += 1
-                    except StopIteration:
-                        logging.warning(
-                            f"There are only {warmup_num} batches in input data, but `INFER_BENCHMARK_WARMUP` has been set to {INFER_BENCHMARK_WARMUP}."
-                        )
-                        break
-                self.benchmark.warmup_stop(warmup_num)
-            output = list(self.apply(input, **kwargs))
-            self.benchmark.collect(len(output))
-        else:
-            yield from self.apply(input, **kwargs)
-
-    def set_predictor(
-        self,
-        batch_size: int = None,
-        device: str = None,
-        pp_option: PaddlePredictorOption = None,
-    ) -> None:
-        """
-        Sets the predictor configuration.
-
-        Args:
-            batch_size (int, optional): The batch size to use. Defaults to None.
-            device (str, optional): The device to run the predictor on. Defaults to None.
-            pp_option (PaddlePredictorOption, optional): The predictor options to set. Defaults to None.
-
-        Returns:
-            None
-        """
-        if batch_size:
-            self.batch_sampler.batch_size = batch_size
-            self.pp_option.batch_size = batch_size
-        if device and device != self.pp_option.device:
-            self.pp_option.device = device
-        if pp_option and pp_option != self.pp_option:
-            self.pp_option = pp_option

paddlex/paddle2onnx_requirements.txt

@@ -1 +0,0 @@
-paddle2onnx>=2

paddlex/repo_manager/requirements.txt

@@ -1,21 +0,0 @@
-tomark~=0.1.4
-psutil~=5.9.2
-wget
-grpcio==1.51.3; python_version<"3.12"
-grpcio==1.59.0; python_version>="3.12"
-pydantic>=2
-editdistance
-openpyxl
-premailer
-python-docx
-######## For Chatocrv3 #######
-langchain==0.2.17
-langchain-openai==0.1.25
-langchain-community==0.2.17
-langchain-text-splitters==0.2.4
-transformers==4.40.0
-openai==1.63.2
-unstructured
-networkx
-numpy==1.24.4; python_version<"3.12"
-numpy==1.26.4; python_version>="3.12"

paddlex/serving_requirements.txt

@@ -1,9 +0,0 @@
-aiohttp>=3.9
-bce-python-sdk>=0.9
-fastapi>=0.110
-filetype>=1.2
-pydantic>=2
-starlette>=0.36
-typing_extensions>=4.11
-uvicorn>=0.16
-yarl>=1.9