docling-ibm-models 3.4.2__tar.gz → 3.4.3__tar.gz

{docling_ibm_models-3.4.2 → docling_ibm_models-3.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: docling-ibm-models
-Version: 3.4.2
+Version: 3.4.3
 Summary: This package contains the AI models used by the Docling PDF conversion package
 License: MIT
 Keywords: docling,convert,document,pdf,layout model,segmentation,table structure,table former
@@ -32,8 +32,9 @@ Requires-Dist: safetensors[torch] (>=0.4.3,<1)
 Requires-Dist: torch (>=2.2.2,<3.0.0)
 Requires-Dist: torchvision (>=0,<1)
 Requires-Dist: tqdm (>=4.64.0,<5.0.0)
-Requires-Dist: transformers (>=4.42.0,<4.43.0) ; sys_platform == "darwin" and platform_machine == "x86_64"
+Requires-Dist: transformers (>=4.42.0,<4.43.0) ; python_version < "3.13" and sys_platform == "darwin" and platform_machine == "x86_64"
 Requires-Dist: transformers (>=4.42.0,<5.0.0) ; sys_platform != "darwin" or platform_machine != "x86_64"
+Requires-Dist: transformers (>=4.47.0,<5.0.0) ; python_version >= "3.13" and (sys_platform != "darwin" or platform_machine != "x86_64")
 Description-Content-Type: text/markdown
 
 [![PyPI version](https://img.shields.io/pypi/v/docling-ibm-models)](https://pypi.org/project/docling-ibm-models/)

{docling_ibm_models-3.4.2 → docling_ibm_models-3.4.3}/docling_ibm_models/code_formula_model/models/sam_opt.py

@@ -82,10 +82,10 @@ class SamOPTModel(OPTModel):
             inputs_embeds = self.embed_tokens(input_ids)
 
         vision_tower = getattr(self, "vision_tower", None)
-        im_start_token = getattr(self.config, "im_start_token", -1)
+        im_start_token = getattr(self.config, "im_start_token", -1)  # type: ignore
 
-        if input_ids.shape[1] != 1 or self.training:
-            with torch.set_grad_enabled(self.training):
+        if input_ids.shape[1] != 1 or self.training:  # type: ignore
+            with torch.set_grad_enabled(self.training):  # type: ignore
                 assert vision_tower is not None
                 image_features = vision_tower(images)
                 image_features = image_features.flatten(2).permute(0, 2, 1)
@@ -118,7 +118,7 @@ class SamOPTModel(OPTModel):
 
             inputs_embeds = torch.stack(new_input_embeds, dim=0)  # type: ignore
 
-        return super(SamOPTModel, self).forward(
+        return super(SamOPTModel, self).forward(  # type: ignore
             input_ids=None,
             attention_mask=attention_mask,
             past_key_values=past_key_values,
@@ -165,12 +165,12 @@ class SamOPTForCausalLM(OPTForCausalLM):
         output_attentions = (
             output_attentions
             if output_attentions is not None
-            else self.config.output_attentions
+            else self.config.output_attentions  # type: ignore
         )
         output_hidden_states = (
             output_hidden_states
             if output_hidden_states is not None
-            else self.config.output_hidden_states
+            else self.config.output_hidden_states  # type: ignore
         )
 
         outputs = self.model(

{docling_ibm_models-3.4.2 → docling_ibm_models-3.4.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docling-ibm-models"
-version = "3.4.2"  # DO NOT EDIT, updated automatically
+version = "3.4.3"  # DO NOT EDIT, updated automatically
 description = "This package contains the AI models used by the Docling PDF conversion package"
 authors = ["Nikos Livathinos <nli@zurich.ibm.com>", "Maxim Lysak <mly@zurich.ibm.com>", "Ahmed Nassar <ahn@zurich.ibm.com>", "Christoph Auer <cau@zurich.ibm.com>", "Michele Dolfi <dol@zurich.ibm.com>", "Peter Staar <taa@zurich.ibm.com>"]
 license = "MIT"
@@ -25,8 +25,9 @@ python = "^3.9"
 torch = "^2.2.2"
 torchvision = "^0"
 transformers = [
+  {markers = "python_version >= '3.13' and (sys_platform != 'darwin' or platform_machine != 'x86_64')", version = "^4.47.0"},
   {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "^4.42.0"},
-  {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~4.42.0"}
+  {markers = "python_version < '3.13' and sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "~4.42.0"}
 ]
 numpy = [
     { version = ">=1.24.4,<3.0.0", markers = 'sys_platform != "darwin" or platform_machine != "x86_64"' },