mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-08 20:58:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add num_tokens as attribtue for VlmPrediction

Browse Source
This commit is contained in:
ElHachem02
2025-10-17 16:41:35 +02:00
parent dd03b53117
commit 311287f562
5 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docling/models/vlm_models_inline/mlx_model.py
View File

@@ -313,5 +313,6 @@ class HuggingFaceMlxModel(BaseVlmPageModel, HuggingFaceModelDownloadMixin):
text=decoded_output,
generation_time=generation_time,
generated_tokens=tokens,
num_tokens = len(tokens)
)
_log.debug("MLX model: Released global lock")