sparrow-parse 0.4.8__py3-none-any.whl → 0.4.9__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- sparrow_parse/__init__.py +1 -1
- sparrow_parse/vllm/mlx_inference.py +6 -5
- {sparrow_parse-0.4.8.dist-info → sparrow_parse-0.4.9.dist-info}/METADATA +1 -1
- {sparrow_parse-0.4.8.dist-info → sparrow_parse-0.4.9.dist-info}/RECORD +7 -7
- {sparrow_parse-0.4.8.dist-info → sparrow_parse-0.4.9.dist-info}/WHEEL +0 -0
- {sparrow_parse-0.4.8.dist-info → sparrow_parse-0.4.9.dist-info}/entry_points.txt +0 -0
- {sparrow_parse-0.4.8.dist-info → sparrow_parse-0.4.9.dist-info}/top_level.txt +0 -0
sparrow_parse/__init__.py
CHANGED
@@ -1 +1 @@
|
|
1
|
-
__version__ = '0.4.
|
1
|
+
__version__ = '0.4.9'
|
@@ -4,8 +4,9 @@ from mlx_vlm.utils import load_image
|
|
4
4
|
from sparrow_parse.vllm.inference_base import ModelInference
|
5
5
|
import os
|
6
6
|
import json
|
7
|
-
from rich import
|
7
|
+
from rich.console import Console
|
8
8
|
|
9
|
+
console = Console()
|
9
10
|
|
10
11
|
class MLXInference(ModelInference):
|
11
12
|
"""
|
@@ -20,7 +21,7 @@ class MLXInference(ModelInference):
|
|
20
21
|
:param model_name: Name of the model to load.
|
21
22
|
"""
|
22
23
|
self.model_name = model_name
|
23
|
-
print(f"MLXInference initialized for model: {model_name}")
|
24
|
+
console.print(f"MLXInference initialized for model: {model_name}")
|
24
25
|
|
25
26
|
|
26
27
|
@staticmethod
|
@@ -32,7 +33,7 @@ class MLXInference(ModelInference):
|
|
32
33
|
:return: Tuple containing the loaded model and processor.
|
33
34
|
"""
|
34
35
|
model, processor = load(model_name)
|
35
|
-
print(f"Loaded model: {model_name}")
|
36
|
+
console.print(f"Loaded model: {model_name}")
|
36
37
|
return model, processor
|
37
38
|
|
38
39
|
|
@@ -53,7 +54,7 @@ class MLXInference(ModelInference):
|
|
53
54
|
formatted_json = json.loads(cleaned_text)
|
54
55
|
return json.dumps(formatted_json, indent=2)
|
55
56
|
except json.JSONDecodeError as e:
|
56
|
-
print(f"Failed to parse JSON in MLX inference backend: {e}")
|
57
|
+
console.print(f"Failed to parse JSON in MLX inference backend: {e}")
|
57
58
|
return output_text
|
58
59
|
|
59
60
|
|
@@ -121,7 +122,7 @@ class MLXInference(ModelInference):
|
|
121
122
|
)
|
122
123
|
results.append(self.process_response(response))
|
123
124
|
|
124
|
-
print("Inference completed successfully for: ", file_path)
|
125
|
+
console.print("Inference completed successfully for: ", file_path)
|
125
126
|
|
126
127
|
return results
|
127
128
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.1
|
2
2
|
Name: sparrow-parse
|
3
|
-
Version: 0.4.
|
3
|
+
Version: 0.4.9
|
4
4
|
Summary: Sparrow Parse is a Python package (part of Sparrow) for parsing and extracting information from documents.
|
5
5
|
Home-page: https://github.com/katanaml/sparrow/tree/main/sparrow-data/parse
|
6
6
|
Author: Andrej Baranovskij
|
@@ -1,4 +1,4 @@
|
|
1
|
-
sparrow_parse/__init__.py,sha256=
|
1
|
+
sparrow_parse/__init__.py,sha256=HVLYuTxwxepBhsoeROAeognEnSW-LkbZ_mSNgbAcTqA,21
|
2
2
|
sparrow_parse/__main__.py,sha256=Xs1bpJV0n08KWOoQE34FBYn6EBXZA9HIYJKrE4ZdG78,153
|
3
3
|
sparrow_parse/extractors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
4
4
|
sparrow_parse/extractors/vllm_extractor.py,sha256=PDLgLlKiq3Bv-UOQTzX3AgxNOLcEU2EniGAXLjMC30U,7820
|
@@ -11,9 +11,9 @@ sparrow_parse/vllm/huggingface_inference.py,sha256=EJnG6PesGKMc_0qGPN8ufE6pSnhAg
|
|
11
11
|
sparrow_parse/vllm/inference_base.py,sha256=4mwGoAY63MB4cHZpV0czTkJWEzimmiTzqqzKmLNzgjw,820
|
12
12
|
sparrow_parse/vllm/inference_factory.py,sha256=FTM65O-dW2WZchHOrNN7_Q3-FlVoAc65iSptuuUuClM,1166
|
13
13
|
sparrow_parse/vllm/local_gpu_inference.py,sha256=aHoJTejb5xrXjWDIGu5RBQWEyRCOBCB04sMvO2Wyvg8,628
|
14
|
-
sparrow_parse/vllm/mlx_inference.py,sha256=
|
15
|
-
sparrow_parse-0.4.
|
16
|
-
sparrow_parse-0.4.
|
17
|
-
sparrow_parse-0.4.
|
18
|
-
sparrow_parse-0.4.
|
19
|
-
sparrow_parse-0.4.
|
14
|
+
sparrow_parse/vllm/mlx_inference.py,sha256=jlycb0WwpQKGkfCrYYgirOdXrqEV6RROB86dmBvAlrU,4961
|
15
|
+
sparrow_parse-0.4.9.dist-info/METADATA,sha256=vunHfk8AaOMZGvmR-5oeph45PBfJMiiYKRbSBGZ-AGA,6432
|
16
|
+
sparrow_parse-0.4.9.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
|
17
|
+
sparrow_parse-0.4.9.dist-info/entry_points.txt,sha256=8CrvTVTTcz1YuZ8aRCYNOH15ZOAaYLlcbYX3t28HwJY,54
|
18
|
+
sparrow_parse-0.4.9.dist-info/top_level.txt,sha256=n6b-WtT91zKLyCPZTP7wvne8v_yvIahcsz-4sX8I0rY,14
|
19
|
+
sparrow_parse-0.4.9.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|