mirror of
https://github.com/opendatalab/MinerU.git
synced 2026-03-27 02:58:54 +07:00
refactor: remove unused import and update help text for device mode option
This commit is contained in:
@@ -4,7 +4,6 @@ import click
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from loguru import logger
|
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.cli_parser import arg_parse
|
||||||
from mineru.utils.config_reader import get_device
|
from mineru.utils.config_reader import get_device
|
||||||
from mineru.utils.guess_suffix_or_lang import guess_suffix_by_path
|
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',
|
'device_mode',
|
||||||
type=str,
|
type=str,
|
||||||
help="""Device mode for model inference, e.g., "cpu", "cuda", "cuda:0", "npu", "npu:0", "mps".
|
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,
|
default=None,
|
||||||
)
|
)
|
||||||
@click.option(
|
@click.option(
|
||||||
|
|||||||
Reference in New Issue
Block a user