optimum-rbln 0.1.13__py3-none-any.whl → 0.2.0__py3-none-any.whl

optimum/rbln/utils/submodule.py

@@ -0,0 +1,137 @@
+# 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 importlib
+from pathlib import Path
+from typing import TYPE_CHECKING, Any, Dict, List
+
+from ..modeling_config import RBLNConfig
+
+
+if TYPE_CHECKING:
+    from transformers import PreTrainedModel
+
+    from ..modeling_base import RBLNBaseModel
+
+
+class SubModulesMixin:
+    """
+    _rbln_submodules = [
+        {"name": "vision_tower"},
+        {"name": "language_model"},
+    ]
+    """
+
+    _rbln_submodules: List[Dict[str, Any]] = []
+
+    def __init__(
+        self,
+        *,
+        rbln_submodules: List["RBLNBaseModel"] = [],
+        **kwargs,
+    ) -> None:
+        for submodule_meta, submodule in zip(self._rbln_submodules, rbln_submodules):
+            setattr(self, submodule_meta["name"], submodule)
+
+    @classmethod
+    def _export_submodules_from_model(
+        cls,
+        model: "PreTrainedModel",
+        model_save_dir: str,
+        rbln_kwargs: Dict[str, Any],
+        **kwargs,
+    ) -> List["RBLNBaseModel"]:
+        rbln_submodules = []
+        for submodule in cls._rbln_submodules:
+            submodule_name = submodule["name"]
+            torch_submodule: "PreTrainedModel" = getattr(model, submodule["name"])
+            cls_name = torch_submodule.__class__.__name__
+            submodule_cls: "RBLNBaseModel" = getattr(importlib.import_module("optimum.rbln"), f"RBLN{cls_name}")
+
+            if submodule_name in rbln_kwargs:
+                kwargs["rbln_config"] = rbln_kwargs[submodule_name]
+
+            rbln_submodule = submodule_cls.from_model(
+                model=torch_submodule,
+                subfolder=submodule_name,
+                model_save_dir=model_save_dir,
+                **kwargs,
+            )
+
+            rbln_submodules.append(rbln_submodule)
+
+        return rbln_submodules
+
+    @classmethod
+    def _load_submodules_from_compiled_models(
+        cls,
+        model_save_dir: str,
+        rbln_kwargs: Dict[str, Any],
+        **kwargs,
+    ):
+        rbln_submodules = []
+        for submodule in cls._rbln_submodules:
+            submodule_name = submodule["name"]
+
+            if submodule_name in rbln_kwargs:
+                kwargs["rbln_config"] = rbln_kwargs[submodule_name]
+
+            # Get cls name for call the constructor of the rbln class
+            submodule_rbln_config = RBLNConfig.load(Path(model_save_dir) / submodule_name)
+            submodule_cls_name = submodule_rbln_config.meta["cls"]
+            submodule_cls: "RBLNBaseModel" = getattr(importlib.import_module("optimum.rbln"), submodule_cls_name)
+
+            rbln_submodule = submodule_cls._from_pretrained(
+                model_id=model_save_dir,
+                config=None,
+                subfolder=submodule_name,
+                **kwargs,
+            )
+            rbln_submodules.append(rbln_submodule)
+        return rbln_submodules
+
+    @classmethod
+    def _load_submodules(
+        cls,
+        model_save_dir,
+        rbln_kwargs,
+        model=None,
+        **kwargs,
+    ):
+        # Two ways :
+        # 1. Compile from pytorch object
+        # 2. Load from compiled file
+        if model is not None:
+            return cls._export_submodules_from_model(
+                model=model,
+                model_save_dir=model_save_dir,
+                rbln_kwargs=rbln_kwargs,
+                **kwargs,
+            )
+
+        else:
+            return cls._load_submodules_from_compiled_models(
+                model_save_dir=model_save_dir,
+                rbln_kwargs=rbln_kwargs,
+                **kwargs,
+            )

optimum_rbln-0.2.0.dist-info/METADATA

@@ -0,0 +1,117 @@
+Metadata-Version: 2.3
+Name: optimum-rbln
+Version: 0.2.0
+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.
+Project-URL: Homepage, https://rebellions.ai
+Project-URL: Documentation, https://docs.rbln.ai
+Author-email: "Rebellions Inc." <support@rebellions.ai>
+License: Apache
+Keywords: atom,diffusers,inference,rbln,rebel,transformers
+Classifier: Development Status :: 2 - Pre-Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Education
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: <3.13,>=3.9
+Requires-Dist: accelerate>=1.0.1
+Requires-Dist: diffusers<=0.31.0
+Requires-Dist: einops>=0.8.0
+Requires-Dist: halo>=0.0.31
+Requires-Dist: packaging>=24.1
+Requires-Dist: torch<=2.5.1
+Requires-Dist: torchaudio<=2.5.1
+Requires-Dist: torchvision<=0.20.1
+Requires-Dist: transformers==4.45.2
+Description-Content-Type: text/markdown
+
+
+# Optimum RBLN
+
+<div align="center">
+
+<img src="assets/rbln_logo.png" width="60%"/>
+
+[![PyPI version](https://badge.fury.io/py/optimum-rbln.svg)](https://badge.fury.io/py/optimum-rbln)
+[![License](https://img.shields.io/github/license/rebellions-sw/optimum-rbln)](https://github.com/rebellions-sw/optimum-rbln/blob/main/LICENSE)
+
+</div>
+
+🤗 Optimum RBLN provides an interface between HuggingFace libraries ([Transformers](https://huggingface.co/docs/transformers), [Diffusers](https://huggingface.co/docs/diffusers/index)) and RBLN NPUs, including [ATOM](https://rebellions.ai/rebellions-product/rbln-ca25/) and [REBEL](https://rebellions.ai/rebellions-product/rebel/).
+
+This library enables seamless integration between the HuggingFace ecosystem and RBLN NPUs through a comprehensive toolkit for model loading and inference across single and multi-NPU environments. While we maintain a list of [officially validated models and tasks](https://docs.rbln.ai/software/optimum/optimum_rbln.html), users can easily adapt other models and tasks with minimal modifications.
+
+## Key Features
+
+🚀 **High Performance Inference**
+- Optimized model execution on RBLN NPUs through RBLN SDK compilation
+- Support for both single and multi-NPU inference
+- Integrated with RBLN Runtime for optimal performance
+
+🔧 **Easy Integration**
+- Seamless compatibility with HuggingFace Model Hub
+- Drop-in replacement for existing HuggingFace pipelines
+- Minimal code changes required for NPU acceleration
+
+
+## Seamless Replacement for Existing HuggingFace Code
+
+```diff
+- from diffusers import StableDiffusionXLPipeline
++ from optimum.rbln import RBLNStableDiffusionXLPipeline
+
+# Load model
+model_id = "stabilityai/stable-diffusion-xl-base-1.0"
+prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+- pipe = StableDiffusionXLPipeline.from_pretrained(model_id)
++ pipe = RBLNStableDiffusionXLPipeline.from_pretrained(model_id, export=True)
+
+# Generate image
+image = pipe(prompt).images[0]
+
+# Save image result
+image.save("image.png")
+
++ # (Optional) Save compiled artifacts to skip the compilation step in future runs
++ pipe.save_pretrained("compiled_sdxl")
+```
+
+## Documentation
+
+Check out [the documentation of Optimum RBLN](https://docs.rbln.ai/software/optimum/optimum_rbln.html) for more advanced usage.
+
+## Getting Started
+
+> **Note:** The `rebel-compiler` library, which is required for running `optimum-rbln`, is only available for approved users. Please refer to the [installation guide](https://docs.rbln.ai/getting_started/installation_guide.html) for instructions on accessing and installing `rebel-compiler`.
+
+### Install from PyPI
+
+To install the latest release of this package:
+```bash
+pip install optimum-rbln
+```
+
+### Install from source
+
+#### Prerequisites
+
+- Install [uv](https://docs.astral.sh/uv/) (refer to [this link](https://docs.astral.sh/uv/getting-started/installation/) for detailed commands)
+
+The below command installs `optimum-rbln` along with its dependencies.
+
+```bash
+git clone https://github.com/rebellions-sw/optimum-rbln.git
+cd optimum-rbln
+./scripts/uv-sync.sh
+```
+
+### Need Help?
+
+- Join discussions and get answers in our [Developer Community](https://discuss.rebellions.ai/)
+- Contact maintainers at [support@rebellions.ai](mailto:support@rebellions.ai)

optimum_rbln-0.2.0.dist-info/RECORD

@@ -0,0 +1,114 @@
+optimum/rbln/__init__.py,sha256=A0ydlEd6-L_ol6wfIFaKl3K93O1j6j9lCO9edPs7MJE,6430
+optimum/rbln/__version__.py,sha256=H-qsvrxCpdhaQzyddR-yajEqI71hPxLa4KxzpP3uS1g,411
+optimum/rbln/modeling.py,sha256=96N6auaSUpxazRNYCOHv2CyC64T8regJrlcMFQGdwqg,8661
+optimum/rbln/modeling_base.py,sha256=3EK3eyv011WK7BXT1vt7chZFpgtiOpdd_gF1uDEX5W8,20753
+optimum/rbln/modeling_config.py,sha256=LxOVAJheEF_A0C3px7pKZCqm3yfTi3-ZGkXJTFBcSc8,11729
+optimum/rbln/diffusers/__init__.py,sha256=YO3hbNDu4jSG1ZL_be8BlQD2IxYH3USBwj71Gdsncnw,3307
+optimum/rbln/diffusers/modeling_diffusers.py,sha256=vml1VgNlJAnYIW1pDTI-IGdph1Ji8ZvBlf4k_rw9864,15322
+optimum/rbln/diffusers/models/__init__.py,sha256=88U_nTn_zDb9lhtT1sn3MSavubrKHu1MmCA48LKOG78,1751
+optimum/rbln/diffusers/models/controlnet.py,sha256=8Hn765YYj5MFs9J4vvseRYkYoH2mYVq6-4VrMsVfUdE,10888
+optimum/rbln/diffusers/models/autoencoders/__init__.py,sha256=yc1ABZG3xxzWPDGf0ADEeuSz3Nrq4ZP-CwddQ-VvWCU,1039
+optimum/rbln/diffusers/models/autoencoders/autoencoder_kl.py,sha256=psW14vL3nJdfTdIdeZ6mF3ed9U85mZN_cX0691ZmnF8,9595
+optimum/rbln/diffusers/models/autoencoders/vae.py,sha256=YdL-vWy9kHRj3fRUjxAnl8eWEYOV2tE0IrxRoIrlkNs,2888
+optimum/rbln/diffusers/models/transformers/__init__.py,sha256=2bVk_6nuqcREOIVyR-4w5ksmdWJqyIV7Wxc5x0dqYO8,1048
+optimum/rbln/diffusers/models/transformers/transformer_sd3.py,sha256=JmqiTq-2iB4y81HrWFAwsGzJISXbU2d6mtH_y7T-_wE,7683
+optimum/rbln/diffusers/models/unets/__init__.py,sha256=-0PyRbBVBFujd7nBh0Z4NOe3RVOlAWyvWLU8r62dqdo,1049
+optimum/rbln/diffusers/models/unets/unet_2d_condition.py,sha256=DrWhElKJ-pMDMfLygmSEm1J0_S_NnnyhqJCHzKW3TuA,14396
+optimum/rbln/diffusers/pipelines/__init__.py,sha256=7rM-JI07v2yQXnb9CLSG0qkTbswzYSPHRYiPraK2Mw4,2863
+optimum/rbln/diffusers/pipelines/controlnet/__init__.py,sha256=k0govvSBxBUR5qpxUGxRMHuQCMX7hXHVZ4EqVRw1LWk,1377
+optimum/rbln/diffusers/pipelines/controlnet/multicontrolnet.py,sha256=l53qsOqazIMnCh71m7RVTaeOlCLdEehKu1Bt48z7Umo,4488
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet.py,sha256=ojuRZEBhMb1tVpqOfGALAX4vS7uoTcvQzlCAJtB0kBo,35557
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py,sha256=L-wKwpgGWTXwpUGqQ5YtxsikwhWiU9FqsU8zsp3Tz8k,34058
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py,sha256=X_CMEJvtXjC46QvmoEIZSt3ErydfVaDwsmyzXiOkhHI,45149
+optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py,sha256=HY8lkm8hZynzrJfPC8Fq3f8sFw_LXZLCi8psKyjzK-8,46478
+optimum/rbln/diffusers/pipelines/stable_diffusion/__init__.py,sha256=7lX6f9XiqROsArw1X0lGsW06H0TrWKwvM9Ye16vis9Y,1224
+optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py,sha256=r17vhwwuK5VQGZbQ62LlgKu1e7LHapBlHZydIkAlQk0,1270
+optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py,sha256=cIW1wJKFZfr5rqHUf7FjKNIN7425w4LgYt5yeth6AuA,1298
+optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py,sha256=rPCj0JsBTrR_XxVh0dpD_8oI3PnYo9ecspGS6t3oYvI,1298
+optimum/rbln/diffusers/pipelines/stable_diffusion_3/__init__.py,sha256=I1IbI4uo2ZqA2uLbhqW67wW7-8DwqeE-qPGc3BTL7dQ,1233
+optimum/rbln/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py,sha256=6ctjMv6MpRuhTle9L9QTaaBuqmeheOBXJ2CgLiG2LA0,1317
+optimum/rbln/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py,sha256=blojGQ8MLbhPAxQfAv9M9fi3UDNrXJGRs_1Uz2Lqa3U,1345
+optimum/rbln/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py,sha256=D_xLTE5D9m_wuQdBmcnIdk48Ey5Q_6mzWH3wex1xt4Q,1345
+optimum/rbln/diffusers/pipelines/stable_diffusion_xl/__init__.py,sha256=eMHZUxczVow6JBNZGZDu_yLbDQIR5EXOGktvsfii2RY,1239
+optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py,sha256=KOZXQLyMkJ4-6ts2SeAJM_8wzKl9kAJ3LBZjqWuCzkI,1296
+optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py,sha256=MT5HwEiua0R7xeApk00YM0vJ_EaUrrqPJUzvmizpdwM,1324
+optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py,sha256=kRd468mTZX4oOKr_9AZH7u7KflbkfQM9bK2CIJ83PBo,1324
+optimum/rbln/ops/__init__.py,sha256=v5MnYLvX82F_mHM2ng8rUprKDbzATt3LnyrPSoMzNeo,1210
+optimum/rbln/ops/attn.py,sha256=UfyzBb760e2FAtFMedQQZFCVjlDtQz7wLehqKRuhjT0,10150
+optimum/rbln/ops/flash_attn.py,sha256=10fc280953kMX6bQpr_0HD5QdzcDUZOnRNz5h56Xmyw,2726
+optimum/rbln/ops/kv_cache_update.py,sha256=skb1fpyzXzVHra1NZJv3wy5PzzaDDhxYD7s3TwGa7RQ,3537
+optimum/rbln/transformers/__init__.py,sha256=gug_v2wgSw_9BYmIk9FFmxnXP7VQ6jFt79zdsySkLpg,4501
+optimum/rbln/transformers/modeling_alias.py,sha256=ezq5taNeBZDOJxA_R9SqiE06bCIxjtLcYDO3nDrQ9Ek,2199
+optimum/rbln/transformers/modeling_generic.py,sha256=InK0I4QWP4w7QrbRPRGJk-pzY4lcBRbxe0sj4sNqIV0,16981
+optimum/rbln/transformers/modeling_rope_utils.py,sha256=ob8haNW5f0tPq26xd4sTg-rMrBT9M2VDP7wxt-PkvYY,13087
+optimum/rbln/transformers/models/__init__.py,sha256=gffOrFFYG3W8ypxpDiUotU-McvjhytffeuUzunjv4nQ,3971
+optimum/rbln/transformers/models/auto/__init__.py,sha256=Dgy4MoIxFxmSwwfTkw75LYGDK9W9kCsGZU4T5qB8qiA,1428
+optimum/rbln/transformers/models/auto/auto_factory.py,sha256=JIFL404RVf6kAONhLeEz6z59tdahcUAyVSU8hdZZe0g,7421
+optimum/rbln/transformers/models/auto/modeling_auto.py,sha256=lUI7gqBJ28cU_-19q1a8xJPwGr07hQhm7lTPi3Q4dko,4327
+optimum/rbln/transformers/models/bart/__init__.py,sha256=F3QY3cwDpHRHxPA0wzzpvxnY1l-p_Xs9yunOZbL0PX4,1068
+optimum/rbln/transformers/models/bart/bart_architecture.py,sha256=5pxQhMvqEugdsXMNv3hreFu5WlsKcMPTMkjJlIYvcLI,5873
+optimum/rbln/transformers/models/bart/modeling_bart.py,sha256=MWtY1qFR0liOcS8JRg53EZ_FtFJeZrAAJT44PXnCtm0,5491
+optimum/rbln/transformers/models/bert/__init__.py,sha256=divBpVNrRAdNAPgnQkGiEZI4oJHCJtLuwdYpMbD3dMM,1034
+optimum/rbln/transformers/models/bert/modeling_bert.py,sha256=akbsBTsGTs7wrxPw120ryZspwYkmHAUrM4A8Kr3COw4,4111
+optimum/rbln/transformers/models/clip/__init__.py,sha256=iXZfPPIztzMDOkY3fbEzx9dCkFKKtWYXCpLGfjEUeZE,1092
+optimum/rbln/transformers/models/clip/modeling_clip.py,sha256=ranLV-NErSUcMEXgR3J4PYaWTMXZaf2qfPabztBLxJM,6099
+optimum/rbln/transformers/models/decoderonly/__init__.py,sha256=lok5Px8lr71TWcLCQhmBBH2qUvwgeBe5nYKUljWAoIk,1059
+optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py,sha256=vLJ_Xq5lukte3PzAxBB32cmkhgFtoPSN0U9n9Lf5XJs,37630
+optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py,sha256=B3ZmKnmBR0xAOSFhNXx1Wm6ahZqO2dh9Zi0CsHjM9xQ,25682
+optimum/rbln/transformers/models/dpt/__init__.py,sha256=R8OHDxOAYPjkk5t8osaPqRh85Pf1Cg1BtzqesqFRcTI,1045
+optimum/rbln/transformers/models/dpt/modeling_dpt.py,sha256=Re15veJwAz3NaUv2GfrHyGUblW0Jcd2KLv23GutFp64,3805
+optimum/rbln/transformers/models/exaone/__init__.py,sha256=CuWNwnZrbd_cLU7jDDPqC0kZIqx1ii_BYyQ98CKDag0,1253
+optimum/rbln/transformers/models/exaone/exaone_architecture.py,sha256=V91GyW2uP3W-DAdBEmm1eN_Qddg2ibOni7qzJaVYqIY,3423
+optimum/rbln/transformers/models/exaone/modeling_exaone.py,sha256=-0VjxpBOQLM3PAmrWCJXkTKQEap577bS_izn-gx8Ew8,2141
+optimum/rbln/transformers/models/gemma/__init__.py,sha256=L1Qfr6dufWtoUZND_ofwxXPSxivEvPR8exy16a_nM2o,1042
+optimum/rbln/transformers/models/gemma/gemma_architecture.py,sha256=ag0TeVt7pkZ3fNDYfi7TpQbwVvObDNvi7kvybtwEGC8,2359
+optimum/rbln/transformers/models/gemma/modeling_gemma.py,sha256=floBHXgogv3iAPyjhjKzbjFHeH67L3FYUKs_wtRm-gM,1924
+optimum/rbln/transformers/models/gpt2/__init__.py,sha256=jsOKYXUclG9G6cwUTUX4eeKqjCPfQUwev7TTFIMXS4Y,1040
+optimum/rbln/transformers/models/gpt2/gpt2_architecture.py,sha256=aYGzhHChNn9UvCRFLe7LwmdmZGNL5b5_x2X__swOXMA,3260
+optimum/rbln/transformers/models/gpt2/modeling_gpt2.py,sha256=lY9mQVEBpymz78Vkk-LpD3sFEJf69GUJaJQ1vH7zhLk,1884
+optimum/rbln/transformers/models/llama/__init__.py,sha256=5mX-MuKzVBj6WQeVxyPhtvFTv0jeZXAFfg4RZ2nVUh0,1042
+optimum/rbln/transformers/models/llama/llama_architecture.py,sha256=UkwDKUmYUtC9_7W4ffjl6M2RNXTuBAwBWVztIbMw3Fg,1122
+optimum/rbln/transformers/models/llama/modeling_llama.py,sha256=D9efkBVUr7TaOkAkiN_qrtQC0AyzLK7cb7UbZpo4XwI,1924
+optimum/rbln/transformers/models/llava_next/__init__.py,sha256=3vi2rmTeKBydGRFOtxELhxWixZggFMpGex6xqfMgi-I,1064
+optimum/rbln/transformers/models/llava_next/modeling_llava_next.py,sha256=vLNI64l90mAzPdwfMuzF67qhY7PBz2gPI10uYUcpFWE,25884
+optimum/rbln/transformers/models/midm/__init__.py,sha256=_6kYchy47frGMZ8uoUspZ9IwrmCBQJ-8kVfXM7xOMew,1249
+optimum/rbln/transformers/models/midm/midm_architecture.py,sha256=W6tYWsRx462CGiB7w4AdTHWUP_ej4ck98a1J3jHhwO8,5696
+optimum/rbln/transformers/models/midm/modeling_midm.py,sha256=9ysJgW76ikr-23XTG_OD-FwM_7uW56fuTOy30wH233U,2128
+optimum/rbln/transformers/models/mistral/__init__.py,sha256=XtuOmzBITjj-H1yctXobJjHF908x1Wlxr_p4hi06v8I,1046
+optimum/rbln/transformers/models/mistral/mistral_architecture.py,sha256=18Tp6h-vnKg524AaH3K56G6tKHMZ6uiyVwUg5cXwAGk,1127
+optimum/rbln/transformers/models/mistral/modeling_mistral.py,sha256=i3X3HKGNee2ocEmpxdHMxuq7UAOgUs-QWlq2OizqA4g,1954
+optimum/rbln/transformers/models/phi/__init__.py,sha256=LrGFTUo1oQnsPSTlxJqAJVVNUdUwq4u_Bf60RUgjLz4,1038
+optimum/rbln/transformers/models/phi/modeling_phi.py,sha256=JfpuUB6cign-lqcUoprgq3gbQclZFT9HGV-NYVkSads,1910
+optimum/rbln/transformers/models/phi/phi_architecture.py,sha256=d71CIMpijgHjGTQbJEFM7JPlmnzkYmiIW_sWLF9HvnY,4390
+optimum/rbln/transformers/models/qwen2/__init__.py,sha256=1PLl1rlF14C6eSk3EZaDfyEHPaC4DZ2vwVlrklTkOYg,1042
+optimum/rbln/transformers/models/qwen2/modeling_qwen2.py,sha256=8ldxWKk85snFX_EViA7kgcgKAZ_QSbmQxhlO4yFvhOA,1924
+optimum/rbln/transformers/models/qwen2/qwen2_architecture.py,sha256=HafB2wFQn4AJ-nHMFc9ftYHZ0WBREHFqucxM6LrRrlA,1114
+optimum/rbln/transformers/models/seq2seq/__init__.py,sha256=Oa11lBWDNQWahqvDco3JIsZldYS-lO8qjpnaGKSfR00,1045
+optimum/rbln/transformers/models/seq2seq/modeling_seq2seq.py,sha256=CGpzCM0wcqzTzbEsw_jA0mGloykaxjHPWQIyMnamZOs,15801
+optimum/rbln/transformers/models/seq2seq/seq2seq_architecture.py,sha256=PrhCrBZb24nzHClQWzllC3-FFulHQ2u7lkbbRX4a1Nc,19447
+optimum/rbln/transformers/models/t5/__init__.py,sha256=ZyK9sxq2_-4Ub6zofOumFUgr8BcCz164RI9yojI98EA,1069
+optimum/rbln/transformers/models/t5/modeling_t5.py,sha256=al83l5tKQ-Xp8rCKTHf4uIWV_VfThpfHLgXK0QEomAo,7987
+optimum/rbln/transformers/models/t5/t5_architecture.py,sha256=u2QyXd9864naTj7aNLlgrt7ILAFzWXvDJJCMsUtEAIE,6603
+optimum/rbln/transformers/models/wav2vec2/__init__.py,sha256=mz4cXqG9b0tDpTAw3qYn3FaJuolX601VmKBE3gohLSw,1043
+optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py,sha256=vRDKoujGRvMvyAlJVJhj_EHf0OtQWbZHickFLzGjmDI,4231
+optimum/rbln/transformers/models/whisper/__init__.py,sha256=PZ8qeAAFMas2MizwVYFxlpFWd5k1Pe1x-0IJfYAMhT8,1059
+optimum/rbln/transformers/models/whisper/generation_whisper.py,sha256=vBtNj7auqjekqwtZQhTIpaauCoFTmufkClz7rRPwZyU,5080
+optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=1C-_mu47aJgd2sDh82mFj030NegYM-_YajZ2d8J_5xM,16215
+optimum/rbln/transformers/models/whisper/whisper_architecture.py,sha256=0JKv4GCDFo_hBL_J_C_AL9UaUTEDE7FGF_zn4b0Q6fk,13552
+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=H5SEtmCAuG6pL1ovl4eLPGZ3tx1IPOilsxKvnbFDN-E,3821
+optimum/rbln/transformers/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+optimum/rbln/transformers/utils/rbln_quantization.py,sha256=_0rnt83lp2Mg6Z6LZvDfznj7em852Yqzy9PTAMFbrzI,7806
+optimum/rbln/utils/__init__.py,sha256=F6hJP00eV1_hT_IVwqqYwLWcLQAvZbmmrNMJTia3mjI,1106
+optimum/rbln/utils/decorator_utils.py,sha256=2yFVaMBDo4F1tHZxZnwcAkhaNm5Q9PXftbSGZWnHXFE,4139
+optimum/rbln/utils/hub.py,sha256=o-nA2I_jnB0S2AX0-q9lSpSNcdP_TeqZbHv84Gxxvi4,4592
+optimum/rbln/utils/import_utils.py,sha256=Vq651F7OpHdf8XUtmCz4jENXIyqCZHmylrn_XhAOFes,4452
+optimum/rbln/utils/logging.py,sha256=on5MA3RRi12SSWgdtHQTiHc0GLRguZafdHaLGdTGGHY,3849
+optimum/rbln/utils/model_utils.py,sha256=Od3XYZyWIm2Kei_BgImSdtqv5zLW9Ilht7lbCWHyfL4,1672
+optimum/rbln/utils/runtime_utils.py,sha256=oGpsxEwI1oEXXCEb3novE2_nRCF3Wh58-G2vk1ukyuI,4093
+optimum/rbln/utils/save_utils.py,sha256=k2PwWStUT-eIY6btyM5A98lMxAAPFas3RxeemET6QI4,4001
+optimum/rbln/utils/submodule.py,sha256=lzD0HYiuo0i5OGnHQgVFdvB4AwLWnPMRCTtMumGY4bc,4465
+optimum_rbln-0.2.0.dist-info/METADATA,sha256=4oNXHUKYeMQGSDpZDqJGcMfTuwmJ_4rCTE2F4vwkiGI,4834
+optimum_rbln-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+optimum_rbln-0.2.0.dist-info/licenses/LICENSE,sha256=q7yy_9ECqQIxXbh2yiG4Hjalfp0e2-2Pbdc7psKUVs8,18079
+optimum_rbln-0.2.0.dist-info/RECORD,,

{optimum_rbln-0.1.13.dist-info → optimum_rbln-0.2.0.dist-info}/WHEEL

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