optimum-rbln 0.7.3a0__py3-none-any.whl → 0.7.3a1__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.
- optimum/rbln/__version__.py +1 -1
- optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py +0 -13
- {optimum_rbln-0.7.3a0.dist-info → optimum_rbln-0.7.3a1.dist-info}/METADATA +1 -1
- {optimum_rbln-0.7.3a0.dist-info → optimum_rbln-0.7.3a1.dist-info}/RECORD +6 -6
- {optimum_rbln-0.7.3a0.dist-info → optimum_rbln-0.7.3a1.dist-info}/WHEEL +0 -0
- {optimum_rbln-0.7.3a0.dist-info → optimum_rbln-0.7.3a1.dist-info}/licenses/LICENSE +0 -0
optimum/rbln/__version__.py
CHANGED
@@ -15,7 +15,6 @@
|
|
15
15
|
import inspect
|
16
16
|
from typing import TYPE_CHECKING, Optional, Union
|
17
17
|
|
18
|
-
import torch
|
19
18
|
from transformers import PretrainedConfig
|
20
19
|
|
21
20
|
from ....modeling import RBLNModel
|
@@ -99,15 +98,3 @@ class RBLNXLMRobertaModel(RBLNModel):
|
|
99
98
|
)
|
100
99
|
rbln_config.model_cfg.update({"max_seq_len": rbln_max_seq_len})
|
101
100
|
return rbln_config
|
102
|
-
|
103
|
-
def forward(
|
104
|
-
self,
|
105
|
-
input_ids: "torch.Tensor",
|
106
|
-
attention_mask: "torch.Tensor",
|
107
|
-
token_type_ids: "torch.Tensor" = None,
|
108
|
-
**kwargs,
|
109
|
-
):
|
110
|
-
if token_type_ids is None:
|
111
|
-
token_type_ids = torch.zeros_like(input=input_ids, dtype=torch.int64)
|
112
|
-
output = super().forward(input_ids, attention_mask, token_type_ids)
|
113
|
-
return output
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.4
|
2
2
|
Name: optimum-rbln
|
3
|
-
Version: 0.7.
|
3
|
+
Version: 0.7.3a1
|
4
4
|
Summary: Optimum RBLN is the interface between the Hugging Face 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
1
|
optimum/rbln/__init__.py,sha256=eHi15YM3989AcX52jka9rUmgAtlp1PHqMNwBEdOfuu8,6554
|
2
|
-
optimum/rbln/__version__.py,sha256=
|
2
|
+
optimum/rbln/__version__.py,sha256=3XXLT-7KoOXBM5ecjGQ9vxdHcJ06x38tTkK1veoUkmQ,513
|
3
3
|
optimum/rbln/modeling.py,sha256=3XE0IrCYbkjw9_Q9BFzZ_ri_Kyxw1g6iwfdohZB46-s,8289
|
4
4
|
optimum/rbln/modeling_base.py,sha256=ELSPbjx7awBRM2SckkD-5gI3TIa01mfzz7gDRC1Pljk,21778
|
5
5
|
optimum/rbln/modeling_config.py,sha256=7104bxmrvKW4Q6XTruQayiIGl8GHDFmPkJ3cknMIInE,11335
|
@@ -102,7 +102,7 @@ optimum/rbln/transformers/models/whisper/generation_whisper.py,sha256=GIHTca3b1V
|
|
102
102
|
optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=0nBADNxE0A1ozBbRutTBvxpo_Y1qkOycT_zronkN-ZU,15840
|
103
103
|
optimum/rbln/transformers/models/whisper/whisper_architecture.py,sha256=eP3UgkwCRaaFjc5Jc4ZEiWxr3-L7oJx9KzpJ7eFkwUs,13158
|
104
104
|
optimum/rbln/transformers/models/xlm_roberta/__init__.py,sha256=fC7iNcdxBZ_6eOF2snStmf8r2M3c8O_-XcXnQEaHQCE,653
|
105
|
-
optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py,sha256=
|
105
|
+
optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py,sha256=8YNLz0bc5ze-QuU8rN-QhUfGzlSUs3iMJiWTxO3o6AM,4366
|
106
106
|
optimum/rbln/transformers/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
107
107
|
optimum/rbln/transformers/utils/rbln_quantization.py,sha256=gwBVHf97sQgPNmGa0wq87E8mPyrtXYhMnO4X4sKp3c8,7639
|
108
108
|
optimum/rbln/utils/__init__.py,sha256=ieDBT2VFTt2E0M4v_POLBpuGW9LxSydpb_DuPd6PQqc,712
|
@@ -114,7 +114,7 @@ optimum/rbln/utils/model_utils.py,sha256=DfD_Z2qvZHqcddXqnzTM1AN8khanj3-DXK2lJvV
|
|
114
114
|
optimum/rbln/utils/runtime_utils.py,sha256=5-DYniyP59nx-mrrbi7AqA77L85b4Cm5oLpaxidSyss,3699
|
115
115
|
optimum/rbln/utils/save_utils.py,sha256=hG5uOtYmecSXZuGTvCXsTM-SiyZpr5q3InUGCCq_jzQ,3619
|
116
116
|
optimum/rbln/utils/submodule.py,sha256=oZoGrItB8WqY4i-K9WJPlLlcLohc1YGB9OHB8_XZw3A,4071
|
117
|
-
optimum_rbln-0.7.
|
118
|
-
optimum_rbln-0.7.
|
119
|
-
optimum_rbln-0.7.
|
120
|
-
optimum_rbln-0.7.
|
117
|
+
optimum_rbln-0.7.3a1.dist-info/METADATA,sha256=59cCm0xXF4GfQ4oMuDLjOYHjkhUMqnZayBGMBstYd0Q,5300
|
118
|
+
optimum_rbln-0.7.3a1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
119
|
+
optimum_rbln-0.7.3a1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
|
120
|
+
optimum_rbln-0.7.3a1.dist-info/RECORD,,
|
File without changes
|
File without changes
|