From 7dd910f067af7a0329f4e404d8aaf4e38e633bd3 Mon Sep 17 00:00:00 2001 From: Yaowei Zheng Date: Wed, 1 Oct 2025 18:20:42 +0800 Subject: [PATCH] [misc] lint (#9221) --- src/llamafactory/chat/chat_model.py | 3 +++ src/llamafactory/cli.py | 7 +++++++ 2 files changed, 10 insertions(+) diff --git a/src/llamafactory/chat/chat_model.py b/src/llamafactory/chat/chat_model.py index b0c9a1cf..9ffd8647 100644 --- a/src/llamafactory/chat/chat_model.py +++ b/src/llamafactory/chat/chat_model.py @@ -49,10 +49,12 @@ class ChatModel: if model_args.infer_backend == EngineName.HF: from .hf_engine import HuggingfaceEngine + self.engine: BaseEngine = HuggingfaceEngine(model_args, data_args, finetuning_args, generating_args) elif model_args.infer_backend == EngineName.VLLM: try: from .vllm_engine import VllmEngine + self.engine: BaseEngine = VllmEngine(model_args, data_args, finetuning_args, generating_args) except ImportError as e: raise ImportError( @@ -62,6 +64,7 @@ class ChatModel: elif model_args.infer_backend == EngineName.SGLANG: try: from .sglang_engine import SGLangEngine + self.engine: BaseEngine = SGLangEngine(model_args, data_args, finetuning_args, generating_args) except ImportError as e: raise ImportError( diff --git a/src/llamafactory/cli.py b/src/llamafactory/cli.py index f7ff80d3..3aa51535 100644 --- a/src/llamafactory/cli.py +++ b/src/llamafactory/cli.py @@ -37,36 +37,43 @@ USAGE = ( def _run_api(): from .api.app import run_api + return run_api() def _run_chat(): from .chat.chat_model import run_chat + return run_chat() def _run_eval(): from .eval.evaluator import run_eval + return run_eval() def _export_model(): from .train.tuner import export_model + return export_model() def _run_exp(): from .train.tuner import run_exp + return run_exp() def _run_web_demo(): from .webui.interface import run_web_demo + return run_web_demo() def _run_web_ui(): from .webui.interface import run_web_ui + return run_web_ui()