mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
00d7d6f123
# Description set up torch_cli using argparses ## Details: - add vllm placeholer in the cli - add unittest for cli command see Readme.md to see how to run the cli Pull Request resolved: https://github.com/pytorch/pytorch/pull/160043 Approved by: https://github.com/huydhn
15 lines
268 B
Python
15 lines
268 B
Python
"""
|
|
Logger Utility helpers for CLI tasks.
|
|
"""
|
|
|
|
import logging
|
|
import sys
|
|
|
|
|
|
def setup_logging(level: int = logging.INFO):
|
|
logging.basicConfig(
|
|
level=level,
|
|
format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
|
|
stream=sys.stdout,
|
|
)
|