Skip to content

Commit ffac295

Browse files
committed
logging when using pipelines_new or models_new
1 parent 044cd01 commit ffac295

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddlex/inference/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,14 +12,16 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from ..utils import flags
15+
from ..utils import logging
1616
from ..utils.flags import USE_NEW_INFERENCE, NEW_PREDICTOR
1717

1818
if USE_NEW_INFERENCE:
19+
logging.warning("=" * 20 + " Using pipelines_new " + "=" * 20)
1920
from .pipelines_new import create_pipeline
2021
else:
2122
from .pipelines import create_pipeline
2223
if NEW_PREDICTOR:
24+
logging.warning("=" * 20 + " Using models_new " + "=" * 20)
2325
from .models_new import create_predictor
2426
else:
2527
from .models import create_predictor

0 commit comments

Comments
 (0)