From 7cfa06503b7587086ac32b8dd3df0ddb08746b0d Mon Sep 17 00:00:00 2001 From: Antonio Jimeno Yepes Date: Thu, 12 Oct 2023 08:01:34 +1100 Subject: [PATCH 1/2] Bug when loading Chipper model --- unstructured_inference/models/chipper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unstructured_inference/models/chipper.py b/unstructured_inference/models/chipper.py index fb172706..3a1d04ea 100644 --- a/unstructured_inference/models/chipper.py +++ b/unstructured_inference/models/chipper.py @@ -46,14 +46,14 @@ class UnstructuredChipperModel(UnstructuredElementExtractionModel): def initialize( self, pre_trained_model_repo: str, - swap_head: bool, - swap_head_hidden_layer_size: int, start_token_prefix: str, prompt: str, max_length: int, heatmap_h: int, heatmap_w: int, source: Source, + swap_head: bool = False, + swap_head_hidden_layer_size: int = 0, no_repeat_ngram_size: int = 10, auth_token: Optional[str] = os.environ.get("UNSTRUCTURED_HF_TOKEN"), device: Optional[str] = None, From 47b9db8cf4b8e20e582ca7bc0484c2fe81e94e35 Mon Sep 17 00:00:00 2001 From: Alan Bertl Date: Wed, 11 Oct 2023 17:44:58 -0500 Subject: [PATCH 2/2] update changelog --- CHANGELOG.md | 2 +- unstructured_inference/__version__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 96f276c0..e054dfee 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,4 +1,4 @@ -## 0.7.3-dev1 +## 0.7.3-dev2 * Integration of Chipperv2 and additional Chipper functionality, which includes automatic detection of GPU, bounding box prediction and hierarchical representation. diff --git a/unstructured_inference/__version__.py b/unstructured_inference/__version__.py index 37b4438d..7e5ecedb 100644 --- a/unstructured_inference/__version__.py +++ b/unstructured_inference/__version__.py @@ -1 +1 @@ -__version__ = "0.7.3-dev1" # pragma: no cover +__version__ = "0.7.3-dev2" # pragma: no cover