optimum-rbln 0.1.4__py3-none-any.whl → 0.1.8__py3-none-any.whl

optimum/rbln/transformers/models/whisper/whisper_architecture.py

@@ -57,7 +57,6 @@ class _WhisperAttention(WhisperAttention):
         cache_position: Optional[torch.Tensor] = None,
         **kwargs,
     ) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
-
         bsz, tgt_len, _ = hidden_states.size()
         is_cross_attention = key_value_states is not None
 
@@ -123,7 +122,6 @@ class _WhisperSdpaAttention(WhisperSdpaAttention):
         cache_position: Optional[torch.Tensor] = None,
         **kwargs,
     ) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
-
         bsz, tgt_len, _ = hidden_states.size()
 
         is_cross_attention = key_value_states is not None
@@ -189,7 +187,6 @@ class _WhisperDecoderLayer(WhisperDecoderLayer):
         cache_position: Optional[torch.Tensor] = None,
         attn_impl: str = "eager",
     ) -> torch.Tensor:
-
         # Self Attention Block
         residual = hidden_states
         hidden_states = self.self_attn_layer_norm(hidden_states)
@@ -248,7 +245,6 @@ class _WhisperDecoder(WhisperDecoder):
         attn_impl: str = "eager",
         **kwargs,
     ):
-
         input_shape = input_ids.size()
         input_ids = input_ids.view(-1, input_shape[-1])
 
@@ -312,7 +308,6 @@ class _WhisperDecoderWrapper(torch.nn.Module):
         self_kv_cache: torch.Tensor,
         cross_kv_cache: torch.Tensor,
     ) -> Union[Tuple[torch.FloatTensor], Seq2SeqLMOutput]:
-
         # prepare past_key_values
         kv_cache = ()
         for i in range(0, self.num_layers * 2, 2):
@@ -367,7 +362,6 @@ class _WhisperEncoderWrapper(torch.nn.Module):
         self,
         input_features: Optional[torch.LongTensor] = None,
     ) -> Union[Tuple[torch.FloatTensor], BaseModelOutput]:
-
         encoder_outputs = self.encoder(input_features=input_features)
         last_hidden_states = encoder_outputs[0]
 

optimum/rbln/transformers/models/xlm_roberta/__init__.py

@@ -0,0 +1,24 @@
+# Copyright 2024 Rebellions Inc.
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+
+#     http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Portions of this software are licensed under the Apache License,
+# Version 2.0. See the NOTICE file distributed with this work for
+# additional information regarding copyright ownership.
+
+# All other portions of this software, including proprietary code,
+# are the intellectual property of Rebellions Inc. and may not be
+# copied, modified, or distributed without prior written permission
+# from Rebellions Inc.
+
+from .modeling_xlm_roberta import RBLNXLMRobertaModel

optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py

@@ -0,0 +1,125 @@
+# Copyright 2024 Rebellions Inc.
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+
+#     http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Portions of this software are licensed under the Apache License,
+# Version 2.0. See the NOTICE file distributed with this work for
+# additional information regarding copyright ownership.
+
+# All other portions of this software, including proprietary code,
+# are the intellectual property of Rebellions Inc. and may not be
+# copied, modified, or distributed without prior written permission
+# from Rebellions Inc.
+
+import logging
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
+
+import torch
+from transformers import AutoModel, PretrainedConfig, PreTrainedModel, XLMRobertaConfig, XLMRobertaModel
+
+from ....modeling_base import RBLNModel
+from ....modeling_config import RBLNConfig, RBLNRuntimeConfig
+
+
+logger = logging.getLogger(__name__)
+
+if TYPE_CHECKING:
+    from transformers import AutoFeatureExtractor, AutoProcessor, AutoTokenizer
+
+class RBLNXLMRobertaModel(RBLNModel):
+    auto_model_class = AutoModel  # feature extraction
+    original_model_class = XLMRobertaModel
+    original_config_class = XLMRobertaConfig
+
+    @classmethod
+    def get_pytorch_model(
+        cls,
+        model_id: str,
+        use_auth_token: Optional[Union[bool, str]] = None,
+        revision: Optional[str] = None,
+        force_download: bool = False,
+        cache_dir: Optional[str] = None,
+        subfolder: str = "",
+        local_files_only: bool = False,
+        trust_remote_code: bool = False,
+        rbln_config_kwargs: Optional[Dict[str, Any]] = None,
+        rbln_constructor_kwargs: Optional[Dict[str, Any]] = None,
+        **kwargs,
+    ) -> "PreTrainedModel":
+        model: "PreTrainedModel" = super().get_pytorch_model(
+            model_id=model_id,
+            use_auth_token=use_auth_token,
+            revision=revision,
+            force_download=force_download,
+            cache_dir=cache_dir,
+            subfolder=subfolder,
+            local_files_only=local_files_only,
+            trust_remote_code=trust_remote_code,
+            rbln_config_kwargs=rbln_config_kwargs,
+            rbln_constructor_kwargs=rbln_constructor_kwargs,
+            library_name="transformers",
+        )
+
+        return model
+
+    @classmethod
+    def _get_rbln_config(
+        cls,
+        preprocessors: Optional[Union["AutoFeatureExtractor", "AutoProcessor", "AutoTokenizer"]],
+        model_config: Optional["PretrainedConfig"] = None,
+        rbln_max_seq_len: Optional[int] = None,
+        rbln_model_input_names: Optional[List[str]] = None,
+        rbln_batch_size: Optional[int] = None,
+    ) -> RBLNConfig:
+
+        max_position_embeddings = getattr(model_config, "n_positions", None) or getattr(
+            model_config, "max_position_embeddings", None
+        )
+
+        if rbln_max_seq_len is None:
+            rbln_max_seq_len = max_position_embeddings
+            if rbln_max_seq_len is None:
+                for tokenizer in preprocessors:
+                    if hasattr(tokenizer, "model_max_length"):
+                        rbln_max_seq_len = tokenizer.model_max_length
+                        break
+                if rbln_max_seq_len is None:
+                    raise ValueError("`rbln_max_seq_len` should be specified!")
+
+        if max_position_embeddings is not None and rbln_max_seq_len > max_position_embeddings:
+            raise ValueError("`rbln_enc_max_seq_len` should be less or equal than max_position_embeddings!")
+
+        if rbln_model_input_names is None:
+            # These are BERT's inputs
+            rbln_model_input_names = ["input_ids", "attention_mask", "token_type_ids"]
+
+        if rbln_batch_size is None:
+            rbln_batch_size = 1
+
+        input_info = [
+            (model_input_name, [rbln_batch_size, rbln_max_seq_len], "int64")
+            for model_input_name in rbln_model_input_names
+        ]
+
+        rbln_runtime_config = RBLNRuntimeConfig(input_info=input_info)
+        rbln_runtime_config.batch_size = rbln_batch_size
+
+        meta = {"rbln_max_seq_len": rbln_max_seq_len}
+
+        return RBLNConfig.from_rbln_runtime_configs([rbln_runtime_config], _rbln_meta=meta)
+
+    def forward(self, input_ids: "torch.Tensor", attention_mask: "torch.Tensor", token_type_ids: "torch.Tensor" = None, **kwargs):
+        if token_type_ids is None:
+            token_type_ids = torch.zeros_like(input=input_ids, dtype=torch.int64)
+        output = super().forward(input_ids, attention_mask, token_type_ids)
+        return output

optimum/rbln/utils/__init__.py

@@ -21,5 +21,5 @@
 # copied, modified, or distributed without prior written permission
 # from Rebellions Inc.
 
-from .import_utils import is_rbln_available
+from .import_utils import check_version_compats, is_rbln_available
 from .runtime_utils import RBLNPytorchRuntime

optimum/rbln/utils/import_utils.py

@@ -21,8 +21,54 @@
 # copied, modified, or distributed without prior written permission
 # from Rebellions Inc.
 
+import importlib.metadata
 import importlib.util
+import warnings
+from dataclasses import dataclass
+
+from packaging.version import Version
+
+
+@dataclass
+class VersionCompat:
+    package_name: str
+    min_version: str
+    max_version: str
+
+
+RBLN_VERSION_COMPATS = {
+    "0.1.5": [
+        VersionCompat(
+            package_name="rebel-compiler",
+            min_version="0.5.7",
+            max_version="0.5.8",
+        ),
+    ],
+    "0.0.0": [],
+}
 
 
 def is_rbln_available() -> bool:
     return importlib.util.find_spec("rebel-compiler") is not None
+
+
+def check_version_compats() -> None:
+    warnings.filterwarnings(action="always", category=ImportWarning)
+
+    my_version = importlib.metadata.version("optimum-rbln")
+    target_version = list(filter(lambda v: Version(my_version) > Version(v), RBLN_VERSION_COMPATS.keys()))[0]
+    for compat in RBLN_VERSION_COMPATS[target_version]:
+        try:
+            dep_version = importlib.metadata.version(compat.package_name)
+        except importlib.metadata.PackageNotFoundError:
+            warnings.warn(f"optimum-rbln requires {compat.package_name} to be installed.", ImportWarning)
+            continue
+
+        if not Version(compat.min_version) <= Version(dep_version) < Version(compat.max_version):
+            warnings.warn(
+                f"optimum-rbln v{my_version} is compatible to {compat.package_name} v{compat.min_version} to v{compat.max_version}. (you are currently using v{dep_version})\n"
+                "Please refer to our SDK release notes at https://docs.rbln.ai/about_atom/release_note.html",
+                ImportWarning,
+            )
+
+    warnings.resetwarnings()

{optimum_rbln-0.1.4.dist-info → optimum_rbln-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: optimum-rbln
-Version: 0.1.4
+Version: 0.1.8
 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.
 Keywords: transformers,diffusers,inference,rbln,atom,rebel
@@ -20,13 +20,13 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Project-URL: Homepage, https://rebellions.ai
 Project-URL: Documentation, https://docs.rbln.ai
 Requires-Python: <3.11,>=3.8
-Requires-Dist: torch==2.2.1
-Requires-Dist: optimum>=1.17.1
+Requires-Dist: torch<=2.2.1
+Requires-Dist: optimum<=1.20.0
 Requires-Dist: accelerate>=0.28.0
-Requires-Dist: transformers==4.40.2
-Requires-Dist: diffusers==0.27.2
+Requires-Dist: transformers<=4.40.2
+Requires-Dist: diffusers<=0.29.2
 Requires-Dist: einops>=0.8.0
-Requires-Dist: diffusers>=0.27.1; extra == "diffusers"
+Requires-Dist: packaging>=24.1
 Requires-Dist: pytest>=8.1.1; extra == "tests"
 Requires-Dist: psutil>=5.9.8; extra == "tests"
 Requires-Dist: parameterized>=0.9.0; extra == "tests"
@@ -34,13 +34,10 @@ Requires-Dist: GitPython>=3.1.42; extra == "tests"
 Requires-Dist: sentencepiece>=0.2.0; extra == "tests"
 Requires-Dist: datasets>=2.18.0; extra == "tests"
 Requires-Dist: sacremoses>=0.1.1; extra == "tests"
-Requires-Dist: diffusers>=0.27.1; extra == "tests"
 Requires-Dist: safetensors>=0.4.2; extra == "tests"
-Requires-Dist: black>=24.3.0; extra == "quality"
 Requires-Dist: ruff>=0.3.3; extra == "quality"
 Requires-Dist: isort>=5.13.2; extra == "quality"
 Requires-Dist: hf-doc-builder>=0.5.0; extra == "quality"
-Provides-Extra: diffusers
 Provides-Extra: tests
 Provides-Extra: quality
 Description-Content-Type: text/markdown
@@ -65,26 +62,28 @@ limitations under the License.
 
 🤗 Optimum RBLN is the interface between the 🤗 Transformers library and RBLN Accelerators including [ATOM](https://atom_link) and [REBEL](https://rebel_link).
 It provides a set of tools enabling easy model loading and inference on single- and multi-Accelerator settings for different downstream tasks.
-The list of officially validated models and tasks is available [here](TODO:). Users can try other models and tasks with only few changes.
+The list of officially validated models and tasks is available [here](https://docs.rbln.ai/software/optimum/optimum_rbln.html). Users can try other models and tasks with only few changes.
 
 ## Install from PyPI
 
 To install the latest release of this package:
 
-```bash
-pip install --index-url https://pypi.rebellions.in/simple optimum-rbln
-```
+- Export environment variables to access to RBLN private PyPI.
+  ```bash
+  export REBEL_PYPI_USERNAME=<username>
+  export REBEL_PYPI_PASSWORD=<password>
+  ```
+
+- Install optimum-rbln package:
+  ```bash
+  pip install --index-url https://pypi.rebellions.in/simple optimum-rbln
+  ```
 
 ## Install from source
 
 ### Prerequisites
 
 - Install [PDM](https://pdm-project.org/latest/) (refer [this link](https://pdm-project.org/latest/#installation) for detailed commands)
-- Export environment variables to access to RBLN private PyPI.
-  ```bash
-  export REBEL_PYPI_USERNAME=<username>
-  export REBEL_PYPI_PASSWORD=<password>
-  ```
 
 The below command installs optimum-rbln along with its dependencies.
 
@@ -106,40 +105,6 @@ pdm install -G:all
 
 🤗 Optimum RBLN was designed with one goal in mind: **to make inference straightforward for any 🤗 Transformers user while leveraging the complete power of RBLN Accelerators**.
 
-#### Transformers Interface
-
-<!--
-There are two main classes one needs to know:
-- TrainiumArgumentParser: inherits the original [HfArgumentParser](https://huggingface.co/docs/transformers/main/en/internal/trainer_utils#transformers.HfArgumentParser) in Transformers with additional checks on the argument values to make sure that they will work well with AWS Trainium instances.
-- [NeuronTrainer](https://huggingface.co/docs/optimum/neuron/package_reference/trainer): this version trainer takes care of doing the proper checks and changes to the supported models to make them trainable on AWS Trainium instances.
-
-The [NeuronTrainer](https://huggingface.co/docs/optimum/neuron/package_reference/trainer) is very similar to the [🤗 Transformers Trainer](https://huggingface.co/docs/transformers/main_classes/trainer), and adapting a script using the Trainer to make it work with Trainium will mostly consist in simply swapping the Trainer class for the NeuronTrainer one.
-That's how most of the [example scripts](https://github.com/huggingface/optimum-neuron/tree/main/examples) were adapted from their [original counterparts](https://github.com/huggingface/transformers/tree/main/examples/pytorch).
-
-```diff
-from transformers import TrainingArguments
-+from optimum.neuron import NeuronTrainer as Trainer
-
-training_args = TrainingArguments(
-  # training arguments...
-)
-
-# A lot of code here
-
-# Initialize our Trainer
-trainer = Trainer(
-    model=model,
-    args=training_args,  # Original training arguments.
-    train_dataset=train_dataset if training_args.do_train else None,
-    eval_dataset=eval_dataset if training_args.do_eval else None,
-    compute_metrics=compute_metrics,
-    tokenizer=tokenizer,
-    data_collator=data_collator,
-)
-``` -->
-
 ### Documentation
 
-Check out [the documentation of Optimum RBLN](https://huggingface.co/docs/optimum-rbln/index) for more advanced usage.
-
-If you find any issue while using those, please open an issue or a pull request.
+Check out [the documentation of Optimum RBLN](https://docs.rbln.ai/software/optimum/optimum_rbln.html) for more advanced usage.

optimum_rbln-0.1.8.dist-info/RECORD

@@ -0,0 +1,73 @@
+optimum/rbln/__init__.py,sha256=3_h6eOK1OuG84FWaN2zSR7FtuUv4YHuzK4_2DsA6uwk,4785
+optimum/rbln/__version__.py,sha256=zemvJ5zjFE6SQT2xmkxc-ZYwNkUTCEX7mz3Epb2qztE,21
+optimum/rbln/diffusers/__init__.py,sha256=w4W7Wy-Mmh8CQZ5M9JnrrE5bN0UsfNehZI41QadE-hk,2605
+optimum/rbln/diffusers/models/__init__.py,sha256=aY6Llq_31dZjdB9HPBDvi7sXVtdQT9r11gokXG5ffxA,1139
+optimum/rbln/diffusers/models/autoencoder_kl.py,sha256=qIhXCfEADNTm2U9I5ZFN1IfA01zwupUY0IBnJwvxLwI,9506
+optimum/rbln/diffusers/models/controlnet.py,sha256=BQVjszdk6qlW6HnSXKRsWR7v6oFNcgMG-utNnRPpLSg,9303
+optimum/rbln/diffusers/models/unet_2d_condition.py,sha256=oZQyT0PpPcxwmDgCSh7d2nT3oU60iKx-AdgJilzWAb8,14468
+optimum/rbln/diffusers/pipelines/__init__.py,sha256=Xr_bQbpbC5HbJB2NuUcVQu2BGebDkc2bhsGJmL6jgps,1449
+optimum/rbln/diffusers/pipelines/controlnet/__init__.py,sha256=k0govvSBxBUR5qpxUGxRMHuQCMX7hXHVZ4EqVRw1LWk,1377
+optimum/rbln/diffusers/pipelines/controlnet/multicontrolnet.py,sha256=FneDaY03zs5aRTeRUTTAbAQsKjGozsea09Km0Z35Ryc,5129
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet.py,sha256=qX0nDFShyZL3RHKgaeeM8XpMLLDsa_PCrhqS2-IfEwM,42605
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py,sha256=9-zw07G5dICByN4k9UgZ5NwPiToRcOApj7M93SM75Pk,41199
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py,sha256=g8SABYroDmcND-0_3CUH1wkdrZPHWdzoWpWLxBk8p-8,53126
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py,sha256=mOlVPumpYbss18jjnZUSwS1EzFjzQDtZf1yOkS4VLng,54485
+optimum/rbln/diffusers/pipelines/stable_diffusion/__init__.py,sha256=qf_uMWSwD-CyRMRC73y1QsTMyl_qCMreIdg0a8rhJuA,1142
+optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py,sha256=rVryl7riAgkkNJzbXQHKRDYEyR7ZhsF_aF_MkMnerco,5399
+optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py,sha256=VfOOybXQnotWIS1ch0a-eMSM-BDrPlZdGwtsYtsH0JQ,5747
+optimum/rbln/diffusers/pipelines/stable_diffusion_xl/__init__.py,sha256=8MDMHIVsDrM6lZAyvpjFtWOFwiY_IoSxzCQe-gJYTPI,159
+optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py,sha256=aCJSoXks7IpjX4rcH6q0TjXtIPzNrbvAvz0KbIEmMr8,5684
+optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py,sha256=Cv9L4El2GOE-3IRQKHNEMuSdWGmtVsRnQJShcv2hOo0,5874
+optimum/rbln/modeling.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+optimum/rbln/modeling_alias.py,sha256=O7vXVl7Qo1tl_UIMMf4z-62pAyKuEZp2x932MTfKoKM,1888
+optimum/rbln/modeling_base.py,sha256=U0YbUG8lcKuZwdvY3BLYouvl-etwmXw2BT1Z3etFeeA,34582
+optimum/rbln/modeling_config.py,sha256=R0GBd-upavJrpR-2SvfgCCxP7f5Zr0NxIqdKmwBfVCk,6439
+optimum/rbln/modeling_seq2seq.py,sha256=XXYu_hpxOgQmMgayUy9URQwgMl9Ci2AfWyVHm7tMP5o,16783
+optimum/rbln/transformers/__init__.py,sha256=OJX33eQx_iOE864MIRpoYCkstc57UtnNiIaoIATp2Zo,2202
+optimum/rbln/transformers/cache_utils.py,sha256=VfArIkAJn3zPXdu-6RXiCWlU-uVwxvhgoMiGxrPtk40,3835
+optimum/rbln/transformers/generation/__init__.py,sha256=6MmqS9D21ir4dcH6_fq8kRsX1VK3QspSn6Qw83F4ORE,1081
+optimum/rbln/transformers/generation/streamers.py,sha256=X-dEmL1L_0Oy0QSFj2RNdamb_xbDWLXd-Ms8ckx6OZ4,5778
+optimum/rbln/transformers/generation/utils.py,sha256=XqPbYQCe6cEuWssG9iHpbbL-wgSQTcyKHabGwKj7lnE,19462
+optimum/rbln/transformers/models/__init__.py,sha256=pBJm_uxcyORi9A9UBdHAu3VdFy59PkuYuVek-31RiLs,1404
+optimum/rbln/transformers/models/bart/__init__.py,sha256=SGUcpQ_5iLsVxySxtbwhRpmGt7BgVUTxHAjxAjQStdU,1063
+optimum/rbln/transformers/models/bart/bart_architecture.py,sha256=T9GjcsL8fAJcvAs_ifnZuDP2F77hhbjBcsc7u53k6OE,14951
+optimum/rbln/transformers/models/clip/__init__.py,sha256=tbco8qW9QhBe3dtWoKgslLZMsXu9dg_KfJ4IgjvK248,1071
+optimum/rbln/transformers/models/clip/modeling_clip.py,sha256=WEXc9LWbClEzxeIw-LliI1A5OpiL7nnDm0G2IzYdqp4,3990
+optimum/rbln/transformers/models/decoderonly/__init__.py,sha256=AG3ib8iZAEDAvVTNhieCyojWZtA67voPB0dI8lbCXTQ,1371
+optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py,sha256=OLN0vvkIlEsP5xQ25x-kk-uw_V201TxWxVMUrGNkW-g,19045
+optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py,sha256=7YfpYWHH3SNCmNYySj7Q-3WSi64pEqMaH1BmDdSYzi8,14191
+optimum/rbln/transformers/models/dpt/__init__.py,sha256=R8OHDxOAYPjkk5t8osaPqRh85Pf1Cg1BtzqesqFRcTI,1045
+optimum/rbln/transformers/models/dpt/modeling_dpt.py,sha256=uN_5DhjGbFmTbpm1JUBgPsDhMP_vIyc0QM2UK5DoRqc,3537
+optimum/rbln/transformers/models/gemma/__init__.py,sha256=L1Qfr6dufWtoUZND_ofwxXPSxivEvPR8exy16a_nM2o,1042
+optimum/rbln/transformers/models/gemma/gemma_architecture.py,sha256=nSLyuw9xMs2f17fVMNQiiuwRp813HP9q2uIyFhnDZ3o,4291
+optimum/rbln/transformers/models/gemma/modeling_gemma.py,sha256=d47ahENZOJ2pNxyZr1ywvnUQhdBjMk5o72rjcqqLz0M,2465
+optimum/rbln/transformers/models/gpt2/__init__.py,sha256=jsOKYXUclG9G6cwUTUX4eeKqjCPfQUwev7TTFIMXS4Y,1040
+optimum/rbln/transformers/models/gpt2/gpt2_architecture.py,sha256=SIzwfQFRNvNOHjZmKcUZiURbnmXNXFh95twYELVmHcg,10278
+optimum/rbln/transformers/models/gpt2/modeling_gpt2.py,sha256=TWfis7dqzPbm23cHfzLl43TwLzyS5nZMcgxSqxBnD1w,5513
+optimum/rbln/transformers/models/llama/__init__.py,sha256=5mX-MuKzVBj6WQeVxyPhtvFTv0jeZXAFfg4RZ2nVUh0,1042
+optimum/rbln/transformers/models/llama/llama_architecture.py,sha256=j4mifSOaIk7wwV9fL9wQSt5kR3rpnvjtxd3VzhMNdgY,1123
+optimum/rbln/transformers/models/llama/modeling_llama.py,sha256=RXv7yyf3dC1h02seAxpRlzrvofgdy9OWE4P8SRwUlk0,2465
+optimum/rbln/transformers/models/midm/__init__.py,sha256=_6kYchy47frGMZ8uoUspZ9IwrmCBQJ-8kVfXM7xOMew,1249
+optimum/rbln/transformers/models/midm/hf_hub_cached/configuration_midm.py,sha256=P5JqTTcx56HOccxKbR14ZjA67BI0RNnJycG738JMaJ4,833
+optimum/rbln/transformers/models/midm/hf_hub_cached/midm_bitext_tokenization.py,sha256=5lhMXfqnIak1PJ9YL-vUxIdY_3DUr3IBXzTqf3ofpmI,12835
+optimum/rbln/transformers/models/midm/hf_hub_cached/modeling_midm.py,sha256=54__wd9EXwGxmaHDksTTcUD2aWl6WoszYsR8dlL1wfE,61031
+optimum/rbln/transformers/models/midm/hf_hub_cached/rotary_position_embedding.py,sha256=5ywaUVKTvqO8GRsHOSXOOGlbiEn-DbGkpJs59_dFb18,4059
+optimum/rbln/transformers/models/midm/midm_architecture.py,sha256=IFnu54MVPFEk5pvaeJ8RZGlCR7X-9MMeTKThXOP6_M0,11367
+optimum/rbln/transformers/models/midm/modeling_midm.py,sha256=y16ALbYdvlpAhXxrXvVmDYk6uc27u79J8SognYVC-VE,5504
+optimum/rbln/transformers/models/t5/__init__.py,sha256=dK6F1jbBf001h79WZiVdiNZoXm5kOe2fskzhREhu0EE,1057
+optimum/rbln/transformers/models/t5/t5_architecture.py,sha256=2nFovfOdiJdY9jdAR9BngwPO3d2Oofn9jqVWgZ-YYZ0,18091
+optimum/rbln/transformers/models/wav2vec2/__init__.py,sha256=mz4cXqG9b0tDpTAw3qYn3FaJuolX601VmKBE3gohLSw,1043
+optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py,sha256=kooQ1CC6p2mHvRHkFE48d69yNTnkG_V6g9Beu6Sy3XU,4063
+optimum/rbln/transformers/models/whisper/__init__.py,sha256=PZ8qeAAFMas2MizwVYFxlpFWd5k1Pe1x-0IJfYAMhT8,1059
+optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=L49ThCv5sqidNevBGsCpGrOSH4H6wzXOCmON1PCmY9M,11996
+optimum/rbln/transformers/models/whisper/whisper_architecture.py,sha256=QtHP5bXUa9HXa95rEdaqsmbH3eG2QJgUBTV1OG7LF-I,15879
+optimum/rbln/transformers/models/xlm_roberta/__init__.py,sha256=NTj4hCpd8L2_i5DZuV5wp-h8OlTLYVUqTrJxzY_Dg9g,1047
+optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py,sha256=krLNMjsmK2I_Ai2JUJzLqE1HIrTDNhpTY2EDQFQV6Kg,4971
+optimum/rbln/utils/__init__.py,sha256=F6hJP00eV1_hT_IVwqqYwLWcLQAvZbmmrNMJTia3mjI,1106
+optimum/rbln/utils/import_utils.py,sha256=i2GmQJC9kl4BvXncVUrqx8VCqfv1omaHiWyCliBxChg,2632
+optimum/rbln/utils/runtime_utils.py,sha256=EzEabg2E18nq2WZRDZWsZ_hgrdgQ7u_NElTMAYpSDvM,2545
+optimum/rbln/utils/save_utils.py,sha256=eFIPtmiblCJ3MvtxEPxmAR3iuLEUrzpyzwtVotDauhw,3283
+optimum_rbln-0.1.8.dist-info/METADATA,sha256=UJKCmPXw-b8Jm1ie2bAWVowLdYQWUuYD-Tz3-WGwksk,4311
+optimum_rbln-0.1.8.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
+optimum_rbln-0.1.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+optimum_rbln-0.1.8.dist-info/RECORD,,

{optimum_rbln-0.1.4.dist-info → optimum_rbln-0.1.8.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: pdm-backend (2.3.1)
+Generator: pdm-backend (2.3.3)
 Root-Is-Purelib: true
 Tag: py3-none-any