mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-26 20:14:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove not-needed function

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-06-01 21:13:54 +02:00
parent 7f6df727e3
commit ffb7f071c3
1 changed files with 0 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
docling/models/hf_vlm_model.py
View File

@ -6,16 +6,6 @@ _log = logging.getLogger(__name__)
class HuggingFaceVlmModel: 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 @staticmethod
def download_models( def download_models(
repo_id: str, repo_id: str,