refactor: remove unused import and update help text for device mode option

This commit is contained in:
myhloli
2025-12-30 11:12:46 +08:00
parent 6f9ef69b34
commit 466b85ba3f

View File

@@ -4,7 +4,6 @@ import click
from pathlib import Path
from loguru import logger
from mineru.utils.check_sys_env import is_mac_os_version_supported
from mineru.utils.cli_parser import arg_parse
from mineru.utils.config_reader import get_device
from mineru.utils.guess_suffix_or_lang import guess_suffix_by_path
@@ -125,7 +124,7 @@ from .common import do_parse, read_fn, pdf_suffixes, image_suffixes
'device_mode',
type=str,
help="""Device mode for model inference, e.g., "cpu", "cuda", "cuda:0", "npu", "npu:0", "mps".
Adapted only for the case where the backend is set to "pipeline" and "vlm-transformers". """,
Adapted only for the case where the backend is set to "pipeline". """,
default=None,
)
@click.option(