From 54cda8ea4240921d9972ae8468d95e94c5b4faf3 Mon Sep 17 00:00:00 2001 From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com> Date: Fri, 1 Sep 2023 09:17:41 +1200 Subject: [PATCH] chore: Change LaMA log statement to use InvokeAI Logger --- invokeai/backend/image_util/lama.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/backend/image_util/lama.py b/invokeai/backend/image_util/lama.py index 2ea22b6fa3..9c5e0153c4 100644 --- a/invokeai/backend/image_util/lama.py +++ b/invokeai/backend/image_util/lama.py @@ -5,6 +5,7 @@ import numpy as np import torch from PIL import Image +import invokeai.backend.util.logging as logger from invokeai.app.services.config import get_invokeai_config from invokeai.backend.util.devices import choose_torch_device @@ -19,7 +20,7 @@ def norm_img(np_img): def load_jit_model(url_or_path, device): model_path = url_or_path - print(f"Loading model from: {model_path}") + logger.info(f"Loading model from: {model_path}") model = torch.jit.load(model_path, map_location="cpu").to(device) model.eval() return model