diff --git a/docling/models/hf_vlm_model.py b/docling/models/hf_vlm_model.py index e82a34d0..20a1c7dd 100644 --- a/docling/models/hf_vlm_model.py +++ b/docling/models/hf_vlm_model.py @@ -6,16 +6,6 @@ _log = logging.getLogger(__name__) class HuggingFaceVlmModel: - @staticmethod - def map_device_to_cpu_if_mlx(device: str) -> str: - if device == "mps": - _log.warning( - "Mapping mlx to cpu for AutoModelForCausalLM, use MLX framework!" - ) - return "cpu" - - return device - @staticmethod def download_models( repo_id: str,