From d5963007f026c959d5d3a8ecc86d55b02b4d07b4 Mon Sep 17 00:00:00 2001 From: kingbri <8082010+kingbri1@users.noreply.github.com> Date: Thu, 8 May 2025 23:45:04 -0400 Subject: [PATCH] Model: Add backend print Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com> --- common/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/common/model.py b/common/model.py index 67af212..9cdfdeb 100644 --- a/common/model.py +++ b/common/model.py @@ -146,7 +146,6 @@ async def load_model_gen(model_path: pathlib.Path, **kwargs): backend_name = unwrap( kwargs.get("backend"), await detect_backend(model_path) ).lower() - print(backend_name) container_class = _BACKEND_REGISTRY.get(backend_name) if not container_class: @@ -162,6 +161,7 @@ async def load_model_gen(model_path: pathlib.Path, **kwargs): f"Available backends: {available_backends}" ) + logger.info(f"Using backend {backend_name}") new_container: BaseModelContainer = await container_class.create( model_path.resolve(), **kwargs )