fix: Error when loading Chipper model (#248)

Author: ajjimeno

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.

unstructured_inference/__version__.py

@@ -1 +1 @@
-__version__ = "0.7.3-dev1"  # pragma: no cover
+__version__ = "0.7.3-dev2"  # pragma: no cover

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,