Skip to content

Commit 82b0cd8

Browse files
authored
chore: change aphrodite ascii art to sonar (#1785)
* change aphrodite ascii art to sonar * auto-pad logo to center model/version
1 parent dc4e975 commit 82b0cd8

2 files changed

Lines changed: 12 additions & 30 deletions

File tree

‎aphrodite/entrypoints/serve/utils/api_utils.py‎

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -306,20 +306,23 @@ def log_version_and_model(lgr: Logger, version: str, model_name: str) -> None:
306306

307307
logo_template = Template(
308308
"\n"
309-
"${w} ▄▄▄ ▄▄▄▄ ▄▄ ▄▄ ▄▄▄▄ ▄▄▄ ▄▄▄▄ ▄▄ ▄▄▄▄▄▄ ▄▄▄▄▄${r}\n"
310-
"${w} ██▀██ ██▄█▀ ██▄██ ██▄█▄ ██▀██ ██▀██ ██ ██ ██▄▄${r}\n"
311-
"${w} ██▀██ ██ ██ ██ ██ ██ ▀███▀ ████▀ ██ ██ ██▄▄▄${r}\n"
309+
"${w} ▗▄▄▖ ▗▄▖ ▗▖ ▗▖ ▗▄▖ ▗▄▄▖${r}\n"
310+
"${w}▐▌ ▐▌ ▐▌▐▛▚▖▐▌▐▌ ▐▌▐▌ ▐▌${r}\n"
311+
"${w} ▝▀▚▖▐▌ ▐▌▐▌ ▝▜▌▐▛▀▜▌▐▛▀▚▖${r}\n"
312+
"${w}▗▄▄▞▘▝▚▄▞▘▐▌ ▐▌▐▌ ▐▌▐▌ ▐▌${r}\n"
312313
)
313314
colors = {"w": "\033[1m", "r": "\033[0m"} # bold default foreground, reset
314315
if formatter != "color":
315316
# monochrome logo (no ansi escape codes)
316317
colors = dict.fromkeys(colors, "")
317318

318-
logo = logo_template.substitute(colors)
319-
width = 54 # logo art width
320-
version_line = f"version: {version}".center(width)
321-
model_line = f"model: {model_name}".center(width)
322-
lgr.info("%s\n%s\n%s\n", logo, version_line, model_line)
319+
version_line = f"version: {version}"
320+
model_line = f"model: {model_name}"
321+
logo_width = max(map(len, logo_template.substitute(w="", r="").splitlines()))
322+
width = max(logo_width, len(version_line), len(model_line))
323+
padding = " " * ((width - logo_width) // 2)
324+
logo = logo_template.substitute(w=padding + colors["w"], r=colors["r"])
325+
lgr.info("%s\n%s\n%s\n", logo, version_line.center(width), model_line.center(width))
323326

324327

325328
async def validate_json_request(raw_request: Request):

‎aphrodite/entrypoints/utils.py‎

Lines changed: 1 addition & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,18 @@
77
import os
88
from argparse import Namespace
99
from http import HTTPStatus
10-
from logging import Logger
1110

1211
import regex as re
1312
from fastapi import Request
1413
from fastapi.responses import JSONResponse, StreamingResponse
1514
from starlette.background import BackgroundTask, BackgroundTasks
1615

17-
from aphrodite import envs
1816
from aphrodite.engine.arg_utils import EngineArgs
1917
from aphrodite.entrypoints.generate.base.protocol import StreamOptions
2018
from aphrodite.entrypoints.openai.models.protocol import LoRAModulePath
2119
from aphrodite.entrypoints.serve.engine.protocol import ErrorInfo, ErrorResponse
2220
from aphrodite.exceptions import GenerationError
23-
from aphrodite.logger import current_formatter_type, init_logger
21+
from aphrodite.logger import init_logger
2422
from aphrodite.platforms import current_platform
2523
from aphrodite.utils.argparse_utils import FlexibleArgumentParser
2624

@@ -257,25 +255,6 @@ def sanitize_message(message: str) -> str:
257255
return message.strip()
258256

259257

260-
def log_version_and_model(lgr: Logger, version: str, model_name: str) -> None:
261-
logo = r"""
262-
▄▄▄ ▄▄▄▄ ▄▄ ▄▄ ▄▄▄▄ ▄▄▄ ▄▄▄▄ ▄▄ ▄▄▄▄▄▄ ▄▄▄▄▄
263-
██▀██ ██▄█▀ ██▄██ ██▄█▄ ██▀██ ██▀██ ██ ██ ██▄▄
264-
██▀██ ██ ██ ██ ██ ██ ▀███▀ ████▀ ██ ██ ██▄▄▄
265-
266-
"""
267-
if envs.APHRODITE_DISABLE_LOG_LOGO or current_formatter_type(lgr) is None:
268-
message = "Aphrodite server version %s, serving model %s"
269-
lgr.info(message, version, model_name)
270-
else:
271-
logo_lines = logo.splitlines()
272-
logo_width = max((len(line.rstrip()) for line in logo_lines if line.strip()), default=0)
273-
version_line = f"version: {version}".center(logo_width)
274-
model_line = f"model: {model_name}".center(logo_width)
275-
message = f"\n{logo}\n{version_line}\n{model_line}\n"
276-
lgr.info("%s", message)
277-
278-
279258
def create_error_response(
280259
message: str | Exception,
281260
err_type: str = "BadRequestError",

0 commit comments

Comments
 (0)