mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
unified logger
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from pydantic import BaseModel, Field # pylint: disable=no-name-in-module
|
||||
from fastapi.exceptions import HTTPException
|
||||
from modules import shared
|
||||
from modules import logger
|
||||
|
||||
|
||||
class ReqFramepack(BaseModel):
|
||||
@@ -60,7 +61,7 @@ def framepack_post(request: ReqFramepack):
|
||||
else:
|
||||
init_image = None
|
||||
except Exception as e:
|
||||
shared.log.error(f"API FramePack: id={task_id} cannot decode init image: {e}")
|
||||
logger.log.error(f"API FramePack: id={task_id} cannot decode init image: {e}")
|
||||
raise HTTPException(status_code=500, detail=str(e)) from e
|
||||
|
||||
try:
|
||||
@@ -69,12 +70,12 @@ def framepack_post(request: ReqFramepack):
|
||||
else:
|
||||
end_image = None
|
||||
except Exception as e:
|
||||
shared.log.error(f"API FramePack: id={task_id} cannot decode end image: {e}")
|
||||
logger.log.error(f"API FramePack: id={task_id} cannot decode end image: {e}")
|
||||
raise HTTPException(status_code=500, detail=str(e)) from e
|
||||
|
||||
del request.init_image
|
||||
del request.end_image
|
||||
shared.log.trace(f"API FramePack: id={task_id} init={init_image.shape} end={end_image.shape if end_image else None} {request}")
|
||||
logger.log.trace(f"API FramePack: id={task_id} init={init_image.shape} end={end_image.shape if end_image else None} {request}")
|
||||
|
||||
generator = run_framepack(
|
||||
_ui_state=None,
|
||||
|
||||
Reference in New Issue
Block a user