optimum-rbln 0.8.3a2__py3-none-any.whl → 0.8.3a3__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.
Potentially problematic release.
This version of optimum-rbln might be problematic. Click here for more details.
- optimum/rbln/__init__.py +4 -0
- optimum/rbln/__version__.py +2 -2
- optimum/rbln/transformers/__init__.py +4 -0
- optimum/rbln/transformers/models/__init__.py +2 -0
- optimum/rbln/transformers/models/depth_anything/__init__.py +16 -0
- optimum/rbln/transformers/models/depth_anything/configuration_depth_anything.py +24 -0
- optimum/rbln/transformers/models/depth_anything/modeling_depth_anything.py +25 -0
- {optimum_rbln-0.8.3a2.dist-info → optimum_rbln-0.8.3a3.dist-info}/METADATA +1 -1
- {optimum_rbln-0.8.3a2.dist-info → optimum_rbln-0.8.3a3.dist-info}/RECORD +11 -8
- {optimum_rbln-0.8.3a2.dist-info → optimum_rbln-0.8.3a3.dist-info}/WHEEL +0 -0
- {optimum_rbln-0.8.3a2.dist-info → optimum_rbln-0.8.3a3.dist-info}/licenses/LICENSE +0 -0
optimum/rbln/__init__.py
CHANGED
|
@@ -81,6 +81,8 @@ _import_structure = {
|
|
|
81
81
|
"RBLNDistilBertForQuestionAnsweringConfig",
|
|
82
82
|
"RBLNDPTForDepthEstimation",
|
|
83
83
|
"RBLNDPTForDepthEstimationConfig",
|
|
84
|
+
"RBLNDepthAnythingForDepthEstimationConfig",
|
|
85
|
+
"RBLNDepthAnythingForDepthEstimation",
|
|
84
86
|
"RBLNExaoneForCausalLM",
|
|
85
87
|
"RBLNExaoneForCausalLMConfig",
|
|
86
88
|
"RBLNGemmaModel",
|
|
@@ -352,6 +354,8 @@ if TYPE_CHECKING:
|
|
|
352
354
|
RBLNDecoderOnlyModelConfig,
|
|
353
355
|
RBLNDecoderOnlyModelForCausalLM,
|
|
354
356
|
RBLNDecoderOnlyModelForCausalLMConfig,
|
|
357
|
+
RBLNDepthAnythingForDepthEstimation,
|
|
358
|
+
RBLNDepthAnythingForDepthEstimationConfig,
|
|
355
359
|
RBLNDistilBertForQuestionAnswering,
|
|
356
360
|
RBLNDistilBertForQuestionAnsweringConfig,
|
|
357
361
|
RBLNDPTForDepthEstimation,
|
optimum/rbln/__version__.py
CHANGED
|
@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
|
|
|
28
28
|
commit_id: COMMIT_ID
|
|
29
29
|
__commit_id__: COMMIT_ID
|
|
30
30
|
|
|
31
|
-
__version__ = version = '0.8.
|
|
32
|
-
__version_tuple__ = version_tuple = (0, 8, 3, '
|
|
31
|
+
__version__ = version = '0.8.3a3'
|
|
32
|
+
__version_tuple__ = version_tuple = (0, 8, 3, 'a3')
|
|
33
33
|
|
|
34
34
|
__commit_id__ = commit_id = None
|
|
@@ -69,6 +69,8 @@ _import_structure = {
|
|
|
69
69
|
"RBLNDistilBertForQuestionAnsweringConfig",
|
|
70
70
|
"RBLNDPTForDepthEstimation",
|
|
71
71
|
"RBLNDPTForDepthEstimationConfig",
|
|
72
|
+
"RBLNDepthAnythingForDepthEstimation",
|
|
73
|
+
"RBLNDepthAnythingForDepthEstimationConfig",
|
|
72
74
|
"RBLNExaoneForCausalLM",
|
|
73
75
|
"RBLNExaoneForCausalLMConfig",
|
|
74
76
|
"RBLNGemmaModel",
|
|
@@ -204,6 +206,8 @@ if TYPE_CHECKING:
|
|
|
204
206
|
RBLNDecoderOnlyModelConfig,
|
|
205
207
|
RBLNDecoderOnlyModelForCausalLM,
|
|
206
208
|
RBLNDecoderOnlyModelForCausalLMConfig,
|
|
209
|
+
RBLNDepthAnythingForDepthEstimation,
|
|
210
|
+
RBLNDepthAnythingForDepthEstimationConfig,
|
|
207
211
|
RBLNDistilBertForQuestionAnswering,
|
|
208
212
|
RBLNDistilBertForQuestionAnsweringConfig,
|
|
209
213
|
RBLNDPTForDepthEstimation,
|
|
@@ -90,6 +90,7 @@ _import_structure = {
|
|
|
90
90
|
"RBLNDecoderOnlyModelForCausalLM",
|
|
91
91
|
"RBLNDecoderOnlyModelForCausalLMConfig",
|
|
92
92
|
],
|
|
93
|
+
"depth_anything": ["RBLNDepthAnythingForDepthEstimationConfig", "RBLNDepthAnythingForDepthEstimation"],
|
|
93
94
|
"dpt": [
|
|
94
95
|
"RBLNDPTForDepthEstimation",
|
|
95
96
|
"RBLNDPTForDepthEstimationConfig",
|
|
@@ -219,6 +220,7 @@ if TYPE_CHECKING:
|
|
|
219
220
|
RBLNDecoderOnlyModelForCausalLM,
|
|
220
221
|
RBLNDecoderOnlyModelForCausalLMConfig,
|
|
221
222
|
)
|
|
223
|
+
from .depth_anything import RBLNDepthAnythingForDepthEstimation, RBLNDepthAnythingForDepthEstimationConfig
|
|
222
224
|
from .distilbert import RBLNDistilBertForQuestionAnswering, RBLNDistilBertForQuestionAnsweringConfig
|
|
223
225
|
from .dpt import RBLNDPTForDepthEstimation, RBLNDPTForDepthEstimationConfig
|
|
224
226
|
from .exaone import RBLNExaoneForCausalLM, RBLNExaoneForCausalLMConfig
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
# Copyright 2025 Rebellions Inc. All rights reserved.
|
|
2
|
+
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at:
|
|
6
|
+
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
|
|
15
|
+
from .configuration_depth_anything import RBLNDepthAnythingForDepthEstimationConfig
|
|
16
|
+
from .modeling_depth_anything import RBLNDepthAnythingForDepthEstimation
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
# Copyright 2025 Rebellions Inc. All rights reserved.
|
|
2
|
+
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at:
|
|
6
|
+
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
|
|
15
|
+
from ...configuration_generic import RBLNModelForDepthEstimationConfig
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
class RBLNDepthAnythingForDepthEstimationConfig(RBLNModelForDepthEstimationConfig):
|
|
19
|
+
"""
|
|
20
|
+
Configuration class for DepthAnythingForDepthEstimation.
|
|
21
|
+
|
|
22
|
+
This configuration class stores the configuration parameters specific to
|
|
23
|
+
RBLN-optimized Depth Anything V2 Small models for depth estimation tasks.
|
|
24
|
+
"""
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
# Copyright 2025 Rebellions Inc. All rights reserved.
|
|
2
|
+
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at:
|
|
6
|
+
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
from ...modeling_generic import RBLNModelForDepthEstimation
|
|
17
|
+
|
|
18
|
+
|
|
19
|
+
class RBLNDepthAnythingForDepthEstimation(RBLNModelForDepthEstimation):
|
|
20
|
+
"""
|
|
21
|
+
RBLN optimized DepthAnythingForDepthEstimation model for depth estimation tasks.
|
|
22
|
+
|
|
23
|
+
This class provides hardware-accelerated inference for Depth Anything V2 Small
|
|
24
|
+
models on RBLN devices, providing the most capable monocular depth estimation (MDE) model.
|
|
25
|
+
"""
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: optimum-rbln
|
|
3
|
-
Version: 0.8.
|
|
3
|
+
Version: 0.8.3a3
|
|
4
4
|
Summary: Optimum RBLN is the interface between the HuggingFace Transformers and Diffusers libraries and RBLN accelerators. It provides a set of tools enabling easy model loading and inference on single and multiple rbln device settings for different downstream tasks.
|
|
5
5
|
Project-URL: Homepage, https://rebellions.ai
|
|
6
6
|
Project-URL: Documentation, https://docs.rbln.ai
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
optimum/rbln/__init__.py,sha256=
|
|
2
|
-
optimum/rbln/__version__.py,sha256=
|
|
1
|
+
optimum/rbln/__init__.py,sha256=uudbkt6M2dI7hyBqnKI7Rw_DcOo6Dye4VdEgHw0Kj2w,17578
|
|
2
|
+
optimum/rbln/__version__.py,sha256=NC494o_z1A8m7nK7GwomeE3oY1ddXSWewbtXIl6IJmQ,712
|
|
3
3
|
optimum/rbln/configuration_utils.py,sha256=xneqnRWSUVROqpzbTrBACex42-L9zwo3eSjfHjFuhv4,33072
|
|
4
4
|
optimum/rbln/modeling.py,sha256=0CMQnpVvW9evNrTFHM2XFbNpRY1HkbFzYJ5sRyYFq0o,14293
|
|
5
5
|
optimum/rbln/modeling_base.py,sha256=gHfqIO6lKT8smkUthUuRHnbITpxHpnDeBPT8iTeasCk,24575
|
|
@@ -72,13 +72,13 @@ optimum/rbln/ops/flash_attn.py,sha256=yTCdYQVqm_1rHMHWjrMQaIR8WTuG_xA6t033x1IVvT
|
|
|
72
72
|
optimum/rbln/ops/kv_cache_update.py,sha256=aIvK2Sp7M3EfJzJgNvIvAJv4emoN6QOhmgaWj-VboLs,1440
|
|
73
73
|
optimum/rbln/ops/linear.py,sha256=5K3pcrrUHu_p8LrMIU-jX2TnafksveFjjZSCsYSp_yw,1328
|
|
74
74
|
optimum/rbln/ops/sliding_window_attn.py,sha256=EQrV_yRGc5z6kvwEsAcLP028bJWkQg2UPI3xubt9skU,3487
|
|
75
|
-
optimum/rbln/transformers/__init__.py,sha256=
|
|
75
|
+
optimum/rbln/transformers/__init__.py,sha256=2Xe086vpwskJzN9bujkuRkBV96Ov54kYmAB7gPjOh-U,11323
|
|
76
76
|
optimum/rbln/transformers/configuration_generic.py,sha256=95ks6REJYuzI1zLwGlPSlxVV45saVcYOob6ihn-WAAY,5092
|
|
77
77
|
optimum/rbln/transformers/modeling_attention_utils.py,sha256=aLyOaq4me1m-JMmnKbuyNQageDxNU2jjEhGE_ew2P5o,11465
|
|
78
78
|
optimum/rbln/transformers/modeling_generic.py,sha256=2BtroigKuu7z7C98dpLwI875R0EoHN-ceHEVbyPQuYk,12212
|
|
79
79
|
optimum/rbln/transformers/modeling_outputs.py,sha256=cd8ZlhHAGq7S6i5-QK6TJCxgORvoPMnZpqPBlUc_pMY,1177
|
|
80
80
|
optimum/rbln/transformers/modeling_rope_utils.py,sha256=6Zg3r-TeUk4WQAlr95pqfhuoAD_RQ4njT1rbO9uPL0Q,14379
|
|
81
|
-
optimum/rbln/transformers/models/__init__.py,sha256=
|
|
81
|
+
optimum/rbln/transformers/models/__init__.py,sha256=IYF1F4TZDfTBlfT3howesi_lNlQe-IUuJFsa4KXmWw4,12126
|
|
82
82
|
optimum/rbln/transformers/models/audio_spectrogram_transformer/__init__.py,sha256=I2vL4lrzbT5p4eJcH-EKHzEfcPkj_XVsie7jb9q6yic,775
|
|
83
83
|
optimum/rbln/transformers/models/audio_spectrogram_transformer/configuration_audio_spectrogram_transformer.py,sha256=z7LJiVJPmnlCM3mcyhPJP8AufSrxO_dsPeJ51onq-Nc,833
|
|
84
84
|
optimum/rbln/transformers/models/audio_spectrogram_transformer/modeling_audio_spectrogram_transformer.py,sha256=FIKEVWpIt6-JQX9B_rAfCrAPqdUHtR2i8D_X2k7639E,1498
|
|
@@ -108,6 +108,9 @@ optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py,sha256=
|
|
|
108
108
|
optimum/rbln/transformers/models/decoderonly/decoderonly_runtime_utils.py,sha256=9acEQxGRzd21YkzxRchkhqxqpX7emQHZigFg60BIulc,19902
|
|
109
109
|
optimum/rbln/transformers/models/decoderonly/generation_decoderonly.py,sha256=4D89IF0yQju_Dp_vLJN_dBkpe2U_LMWaUciYx57D-0M,3379
|
|
110
110
|
optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py,sha256=hu2eJr0CpLHnRPSLhyBhyyC6DfosKmPu7lPjapcBCkE,33061
|
|
111
|
+
optimum/rbln/transformers/models/depth_anything/__init__.py,sha256=xvPSIriMJWyNeVYoVB1Z7YqB4kkHOIkaHq7loNps-dk,756
|
|
112
|
+
optimum/rbln/transformers/models/depth_anything/configuration_depth_anything.py,sha256=JujBVEUa_zZDXNPr1y-B_PhK5SgFFcY8Ib4EoGjjtmE,989
|
|
113
|
+
optimum/rbln/transformers/models/depth_anything/modeling_depth_anything.py,sha256=ganUtyPKcdKe5QCJ2atQhblzfAstyYkThYDm_DIecU8,1014
|
|
111
114
|
optimum/rbln/transformers/models/distilbert/__init__.py,sha256=zXL78SOEORTnUN_wrdoaDaYpntG8lcFHvPobM6jC0CI,841
|
|
112
115
|
optimum/rbln/transformers/models/distilbert/configuration_distilbert.py,sha256=O3BW9JjyYk9PLyiofvOKEgTdMZ_jpIuPfot281pSsyg,984
|
|
113
116
|
optimum/rbln/transformers/models/distilbert/modeling_distilbert.py,sha256=LUh6zYGa8AR3Yxaj3gtyJRc-czBN3qnHTc-JTAhuqY0,1099
|
|
@@ -227,7 +230,7 @@ optimum/rbln/utils/model_utils.py,sha256=4k5879Kh75m3x_vS4-qOGfqsOiAvc2kdNFFfvsF
|
|
|
227
230
|
optimum/rbln/utils/runtime_utils.py,sha256=R6uXDbeJP03-FWdd4vthNe2D4aCra5n12E3WB1ifiGM,7933
|
|
228
231
|
optimum/rbln/utils/save_utils.py,sha256=hG5uOtYmecSXZuGTvCXsTM-SiyZpr5q3InUGCCq_jzQ,3619
|
|
229
232
|
optimum/rbln/utils/submodule.py,sha256=w5mgPgncI740gVKMu3S-69DGNdUSI0bTZxegQGcZ98Y,5011
|
|
230
|
-
optimum_rbln-0.8.
|
|
231
|
-
optimum_rbln-0.8.
|
|
232
|
-
optimum_rbln-0.8.
|
|
233
|
-
optimum_rbln-0.8.
|
|
233
|
+
optimum_rbln-0.8.3a3.dist-info/METADATA,sha256=NmmTKyVMLYeulq_6EaNrsurXob_0dJ59oBbz1Eeh2VA,5299
|
|
234
|
+
optimum_rbln-0.8.3a3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
|
235
|
+
optimum_rbln-0.8.3a3.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
|
|
236
|
+
optimum_rbln-0.8.3a3.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|