optimum-rbln 0.8.2a0__py3-none-any.whl → 0.8.2a2__py3-none-any.whl

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

@@ -345,12 +345,14 @@ class RBLNWhisperForConditionalGeneration(RBLNModel, RBLNWhisperGenerationMixin)
                 tensor_type="pt",
                 device=rbln_config.device_map["encoder"],
                 activate_profiler=rbln_config.activate_profiler,
+                timeout=rbln_config.timeout,
             ),
             rebel.Runtime(
                 compiled_models[1],
                 tensor_type="pt",
                 device=rbln_config.device_map["decoder"],
                 activate_profiler=rbln_config.activate_profiler,
+                timeout=rbln_config.timeout,
             ),
         ]
 

optimum/rbln/utils/runtime_utils.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import re
 import threading
 from typing import Any, Dict, List, Optional, Union
 
@@ -19,6 +20,21 @@ import rebel
 import torch
 
 
+def normalize_npu(npu: str) -> str:
+    """Normalize the NPU string by removing the form factor."""
+    match = re.match(r"(RBLN-CA|RBLN-CR)(\d+)", npu)
+    if match:
+        prefix, num = match.groups()
+        if len(num) == 1:
+            # Convert "RBLN-CAx" → "RBLN-CA0"
+            # (e.g., "RBLN-CA2" -> "RBLN-CA0")
+            npu = f"{prefix}0"
+        elif len(num) == 2:
+            # Strip form factor (e.g., "RBLN-CA15" → "RBLN-CA1")
+            npu = f"{prefix}{num[:-1]}"
+    return npu
+
+
 def tp_and_devices_are_ok(
     tensor_parallel_size: Optional[int] = None,
     device: Optional[Union[int, List[int]]] = None,
@@ -58,7 +74,7 @@ def tp_and_devices_are_ok(
     if npu is not None:
         for device_id in device:
             npu_name = rebel.get_npu_name(device_id)
-            if npu_name != npu:
+            if normalize_npu(npu_name) != normalize_npu(npu):
                 return f"Device {device_id} ({npu_name}) is not on the same NPU as {npu}."
 
     return None
@@ -147,13 +163,20 @@ class ContextRblnConfig:
     _local = threading.local()
 
     def __init__(
-        self, device=None, device_map=None, create_runtimes=None, optimize_host_mem=None, activate_profiler=None
+        self,
+        device=None,
+        device_map=None,
+        create_runtimes=None,
+        optimize_host_mem=None,
+        activate_profiler=None,
+        timeout=None,
     ):
         self.device = device
         self.device_map = device_map
         self.create_runtimes = create_runtimes
         self.optimize_host_mem = optimize_host_mem
         self.activate_profiler = activate_profiler
+        self.timeout = timeout
 
     def __enter__(self):
         self._local.device = self.device
@@ -161,6 +184,7 @@ class ContextRblnConfig:
         self._local.create_runtimes = self.create_runtimes
         self._local.optimize_host_memory = self.optimize_host_mem
         self._local.activate_profiler = self.activate_profiler
+        self._local.timeout = self.timeout
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
@@ -169,6 +193,7 @@ class ContextRblnConfig:
         self._local.create_runtimes = None
         self._local.optimize_host_memory = None
         self._local.activate_profiler = None
+        self._local.timeout = None
 
     @classmethod
     def get_current_context(cls):
@@ -178,4 +203,5 @@ class ContextRblnConfig:
             "create_runtimes": getattr(cls._local, "create_runtimes", None),
             "optimize_host_memory": getattr(cls._local, "optimize_host_memory", None),
             "activate_profiler": getattr(cls._local, "activate_profiler", None),
+            "timeout": getattr(cls._local, "timeout", None),
         }

{optimum_rbln-0.8.2a0.dist-info → optimum_rbln-0.8.2a2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimum-rbln
-Version: 0.8.2a0
+Version: 0.8.2a2
 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.
 Project-URL: Homepage, https://rebellions.ai
 Project-URL: Documentation, https://docs.rbln.ai

{optimum_rbln-0.8.2a0.dist-info → optimum_rbln-0.8.2a2.dist-info}/RECORD

@@ -1,17 +1,17 @@
-optimum/rbln/__init__.py,sha256=MZCYmY4Y_Zfk0TGo3xK52osHDLZHz4cSdduXZt6RfSI,15316
-optimum/rbln/__version__.py,sha256=kKne35dFUj-l3bjR0tLZka8O-dDdB-rFDsjhN13A2r4,519
-optimum/rbln/configuration_utils.py,sha256=o5oer7fBdE-MHLGNXoP35FjmuQbMmjEIDv0QE_k3kpo,32336
-optimum/rbln/modeling.py,sha256=bsvK6GQtoH9vx72Ea59kvv61jguOk9XDTzVjsY1ugkk,14248
+optimum/rbln/__init__.py,sha256=Pl_On5dBWYbmJk6776cc0eU49oc0pebdCvgb2l4hQRA,15564
+optimum/rbln/__version__.py,sha256=0WA5HC48CQUocr8qaZZFi2jxgajnt6fwHhbsacDdi4g,519
+optimum/rbln/configuration_utils.py,sha256=tnR9XZfzZ9oHc7nU_kX33oo3qFFyicICSVQSujYPiOM,32911
+optimum/rbln/modeling.py,sha256=wZ_Cw83tgPDDJAB84_iLEB_H7ycL-rrlTsCzrHk-GWo,14293
 optimum/rbln/modeling_base.py,sha256=QpNkU_Do__JKmnHjaPzv47OhQwgGfVohisip1jqXa7A,23871
 optimum/rbln/diffusers/__init__.py,sha256=_3FoA0uxCdFd6YK9PMaptFmR9XvdB_PcvYR1MkbGpN8,6957
-optimum/rbln/diffusers/modeling_diffusers.py,sha256=RjZNcYMU5daUIj-PAxyAwVoo2a9hnbR81qfcwUYJBPI,19844
+optimum/rbln/diffusers/modeling_diffusers.py,sha256=0Pwqw2PbLSQ6FXsXzbx44Jfz3Vm9hfSrfVRH5zcN0yk,19885
 optimum/rbln/diffusers/configurations/__init__.py,sha256=vMRnPY4s-Uju43xP038D2EA18X_mhy2YfsZVpSU-VoA,1322
 optimum/rbln/diffusers/configurations/models/__init__.py,sha256=7q95gtgDzCeIBogGw8SLQoHT4Wch7vpLJVF2UQovuoo,567
 optimum/rbln/diffusers/configurations/models/configuration_autoencoder_kl.py,sha256=ySetuNq6koleFIZ542zZLTzEEyl_CTul9l12ufWlQ_Y,3218
 optimum/rbln/diffusers/configurations/models/configuration_autoencoder_kl_cosmos.py,sha256=SJfgbUz1LlRVuKQ_sHwPS262oOHF2TliKqM2z13wjEw,4172
 optimum/rbln/diffusers/configurations/models/configuration_controlnet.py,sha256=VDO_YFS_QhcHhuRIXQL53JZXEO27yoKHtecq5hd2la8,2637
 optimum/rbln/diffusers/configurations/models/configuration_prior_transformer.py,sha256=vE8RsXc27Z4-9k0KEM_vP7AWd5UUYvDgfX1g6nUrPp4,2224
-optimum/rbln/diffusers/configurations/models/configuration_transformer_cosmos.py,sha256=tqzBWzkl5PX60v8REGHuUC1WdJuIQv_2BGUOne5UYL8,3127
+optimum/rbln/diffusers/configurations/models/configuration_transformer_cosmos.py,sha256=0FAXTBJwS1zmlYkeChk8ndVVyd_ozjO3x8RN3MAyEsI,3205
 optimum/rbln/diffusers/configurations/models/configuration_transformer_sd3.py,sha256=TAwHUyVy_9HSEZdXIuFCtrBfNIuYIedklJaCut5wEys,2412
 optimum/rbln/diffusers/configurations/models/configuration_unet_2d_condition.py,sha256=mxcbrOqLMnPpP-jnjSeRWPj2zwPMsgeQSq6LzhG2btA,3630
 optimum/rbln/diffusers/configurations/models/configuration_vq_model.py,sha256=dslGcfCZL_hNeVyjV-4FnCT1POmXuiaLbr6NcQSKgHg,3259
@@ -25,13 +25,13 @@ optimum/rbln/diffusers/configurations/pipelines/configuration_stable_diffusion_x
 optimum/rbln/diffusers/models/__init__.py,sha256=TgrKGOPDJyEtDLiF22L1q1M8tCal4D9wBcrAtE5Zs5Y,1624
 optimum/rbln/diffusers/models/controlnet.py,sha256=6owledPe9BXhbZOG8lbuuYvpBU0UrQV7zmat6SoMXOM,10585
 optimum/rbln/diffusers/models/autoencoders/__init__.py,sha256=kpINW4bWwy-Q2doPME4nZ8gXRmkK2DRv2kDdbZuQ3m8,738
-optimum/rbln/diffusers/models/autoencoders/autoencoder_kl.py,sha256=W5dAVtl7y3Ho63Zq9g-CG16_odku3uZ8fsD8IOePpgI,8534
-optimum/rbln/diffusers/models/autoencoders/autoencoder_kl_cosmos.py,sha256=aHkpiMemMU9Yc0dKCt90RBEGlcGzdkO3pWip-w1itXM,8669
+optimum/rbln/diffusers/models/autoencoders/autoencoder_kl.py,sha256=YIKZicC8elQmT_ERYai4xXjetcJLzlngYZsNI0NT39M,8579
+optimum/rbln/diffusers/models/autoencoders/autoencoder_kl_cosmos.py,sha256=nGoiEJj5Xnat6isFr7vZIw8Ff2SVzoPdEqL-UbB7qXA,8714
 optimum/rbln/diffusers/models/autoencoders/vae.py,sha256=ZX6XH9eZxexSSN8sSKFDcvEK9mMuEQNyoalSpOOqQrE,5419
-optimum/rbln/diffusers/models/autoencoders/vq_model.py,sha256=IA7D9c1SMnsZpDJ-xO85cbmXznA-vKhZkWwHZP0b3sw,6739
+optimum/rbln/diffusers/models/autoencoders/vq_model.py,sha256=TgeLxomob9AShWHEpchwTJGqInNV1cj5clkMKmJvDiM,6784
 optimum/rbln/diffusers/models/transformers/__init__.py,sha256=3oTqAOok-dUR2KealC41CKt36dpKq3IT4kabmHkrCpg,767
 optimum/rbln/diffusers/models/transformers/prior_transformer.py,sha256=SWoeVK--BRMwuXVABNVtonmzJDusx0iz4Q3EAvJ9uN8,5395
-optimum/rbln/diffusers/models/transformers/transformer_cosmos.py,sha256=_YwjOqIGluIt-Nvax8_YJb7BtyZcSf9OAJoYSZsVB2I,12704
+optimum/rbln/diffusers/models/transformers/transformer_cosmos.py,sha256=UQ_R7RVJ4epreJ59U3kAuuVbH-InYvRH0cvoB9JYTco,12720
 optimum/rbln/diffusers/models/transformers/transformer_sd3.py,sha256=yF7sS0QvawowpV9hR5GeT8DaE8CCp3mj1njHHd9cKTc,6630
 optimum/rbln/diffusers/models/unets/__init__.py,sha256=MaICuK9CWjgzejXy8y2NDrphuEq1rkzanF8u45k6O5I,655
 optimum/rbln/diffusers/models/unets/unet_2d_condition.py,sha256=v3WS9EGKROE_QClXrxC7rmRko1BspAvAbeIfh83LK88,15832
@@ -43,8 +43,8 @@ optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py,sha25
 optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py,sha256=2w6dmGQuBWqVoocn27z2yMkG7fL7_MVDBcQNJPJsRXU,45300
 optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py,sha256=HX56itORMqXLjZcwv25C-_z3JyZn3v6BpfIjsrDO3mE,46640
 optimum/rbln/diffusers/pipelines/cosmos/__init__.py,sha256=h2j6S8IJPVHeNU8qmW9vyXMgHBw0d7kQcuMAA5YoHPU,795
-optimum/rbln/diffusers/pipelines/cosmos/configuration_cosmos_guardrail.py,sha256=kDVnUBBGdumpDj2DaOpo5MSsFvlFIGY6BU1LZaFVqao,3327
-optimum/rbln/diffusers/pipelines/cosmos/cosmos_guardrail.py,sha256=EAt2UICPRTaUz4SNsQYOa9aoW0USj2qamqdhlf2ajrA,18261
+optimum/rbln/diffusers/pipelines/cosmos/configuration_cosmos_guardrail.py,sha256=Q3cELqGuSbubcHNY-v2BSt84xy8BTvryZrhynhBBvjA,3559
+optimum/rbln/diffusers/pipelines/cosmos/cosmos_guardrail.py,sha256=s7S-g2ZpomwzqO80OJdwZr8m3VTtqrBnuMgDOsXFn_Y,16880
 optimum/rbln/diffusers/pipelines/cosmos/pipeline_cosmos_text2world.py,sha256=TfhgAWVHUHvxsagBGLAVYKBoSMvuH7rg_xP5ZZ0rVU0,3910
 optimum/rbln/diffusers/pipelines/cosmos/pipeline_cosmos_video2world.py,sha256=-dl8AMwSuorIOxRNfyu1XhkJfmNVbSo3_Wkb2gAmUpo,3917
 optimum/rbln/diffusers/pipelines/kandinsky2_2/__init__.py,sha256=I4YQq2HfA3xONbWsdJ870IEJPyLWeCDDG-UCJsu9YO8,1035
@@ -71,11 +71,11 @@ optimum/rbln/ops/flash_attn.py,sha256=z39DJZSk94630ueoOCkiybxR5gzvNR-SRADHs0F6pz
 optimum/rbln/ops/kv_cache_update.py,sha256=HjnHBR-oFrJQibsVnkYb0P5_-wEma8jl0mkjkylwakU,1270
 optimum/rbln/ops/linear.py,sha256=1_7Hg-9wXxhu97fqPobotLQx17k7VPeSSL91_9Z7EDg,1018
 optimum/rbln/ops/sliding_window_attn.py,sha256=EQrV_yRGc5z6kvwEsAcLP028bJWkQg2UPI3xubt9skU,3487
-optimum/rbln/transformers/__init__.py,sha256=MF7OaGf-KI9rz4EOzejxHTDYUB3RO2L02BquTe0PXmI,9107
+optimum/rbln/transformers/__init__.py,sha256=6IClRtf05wYmJSnD3-2xMDH5k79bJZeOarJhaFcLmvM,9355
 optimum/rbln/transformers/configuration_generic.py,sha256=kNhPWtzF0IovUnrsXfxXdXITqgpfCAAedjfB6jSAhEg,5131
 optimum/rbln/transformers/modeling_generic.py,sha256=SXsZghRDsPolNnG5FFPRtXzIEmPQDnz0iRy6PIZvFVI,12225
 optimum/rbln/transformers/modeling_rope_utils.py,sha256=6Zg3r-TeUk4WQAlr95pqfhuoAD_RQ4njT1rbO9uPL0Q,14379
-optimum/rbln/transformers/models/__init__.py,sha256=VVQJgpUUnN4MPAQlOsxsw63w7WPK05ggFfRkGYuZFJQ,10266
+optimum/rbln/transformers/models/__init__.py,sha256=8wziKlMxcLROSNN697jZItjdM0gNSrcpQN_FsLrL26o,10487
 optimum/rbln/transformers/models/audio_spectrogram_transformer/__init__.py,sha256=I2vL4lrzbT5p4eJcH-EKHzEfcPkj_XVsie7jb9q6yic,775
 optimum/rbln/transformers/models/audio_spectrogram_transformer/configuration_audio_spectrogram_transformer.py,sha256=z7LJiVJPmnlCM3mcyhPJP8AufSrxO_dsPeJ51onq-Nc,833
 optimum/rbln/transformers/models/audio_spectrogram_transformer/modeling_audio_spectrogram_transformer.py,sha256=FIKEVWpIt6-JQX9B_rAfCrAPqdUHtR2i8D_X2k7639E,1498
@@ -101,8 +101,8 @@ optimum/rbln/transformers/models/colpali/configuration_colpali.py,sha256=ieY-tuy
 optimum/rbln/transformers/models/colpali/modeling_colpali.py,sha256=jzvJCBrrCXSpjfmJ3O-VvPNFGWGaNbpOV09JwLPAZWs,15757
 optimum/rbln/transformers/models/decoderonly/__init__.py,sha256=vQYZDDdoddwA7yKc5zzrq2Zs9sax-0p8rNF_aYfF4bk,1006
 optimum/rbln/transformers/models/decoderonly/configuration_decoderonly.py,sha256=cakn8RGo8gS3nmXdEqOfC2xUBOMGInROgLEbCOoLFR0,13398
-optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py,sha256=8ovJ5_q_asqVTuVnAuK1m6genW0OSJ30Cd7HS9JXJgc,46363
-optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py,sha256=NmWdodIcXXChI61-Ej7StTe52iQvalMYRUDuNtcQVEI,53342
+optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py,sha256=HrI12t9X9wV_-AZuTBSs-W7c5yVUkvd0secWlI72x2A,46325
+optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py,sha256=qe0icOCkPY1k1Db9gP5fXu-kHIjZV3M_tOjrs-EeULQ,53701
 optimum/rbln/transformers/models/distilbert/__init__.py,sha256=zXL78SOEORTnUN_wrdoaDaYpntG8lcFHvPobM6jC0CI,841
 optimum/rbln/transformers/models/distilbert/configuration_distilbert.py,sha256=O3BW9JjyYk9PLyiofvOKEgTdMZ_jpIuPfot281pSsyg,984
 optimum/rbln/transformers/models/distilbert/modeling_distilbert.py,sha256=LUh6zYGa8AR3Yxaj3gtyJRc-czBN3qnHTc-JTAhuqY0,1099
@@ -120,7 +120,7 @@ optimum/rbln/transformers/models/gemma/modeling_gemma.py,sha256=Ojvum34EhDHWfMB4
 optimum/rbln/transformers/models/gemma3/__init__.py,sha256=6rugk3615SEt4lh7gduo_J9VyGiSReuEIvL0Uno0eaI,790
 optimum/rbln/transformers/models/gemma3/configuration_gemma3.py,sha256=eupMGTHJGJNNrAZ3GE6M6GQBAQzBb7KFJvalyDmbM-A,3063
 optimum/rbln/transformers/models/gemma3/gemma3_architecture.py,sha256=fpLDAXCe5paWVsfc0tL59JkRQMRF-WNgIzOIb_QpSLU,6191
-optimum/rbln/transformers/models/gemma3/modeling_gemma3.py,sha256=-cpU0ot46VFUZ6PtfwN9VJ-E44n6mP1E3dKwB99MtBM,38389
+optimum/rbln/transformers/models/gemma3/modeling_gemma3.py,sha256=iVzH7dm4ZO1raw6mznhmTkCY4UYbTzI5lm-8X7I1vN0,38528
 optimum/rbln/transformers/models/gpt2/__init__.py,sha256=socBMIBZSiLbrVN12rQ4nL9gFeT0axMgz6SWaCaD4Ac,704
 optimum/rbln/transformers/models/gpt2/configuration_gpt2.py,sha256=9sS6-EGapmow3rG9ViejK9qwrqy_X86VBxQ7u9x0Yqk,923
 optimum/rbln/transformers/models/gpt2/gpt2_architecture.py,sha256=O7hBiaFJrpLSswGwW83cX9S9Q2wKRBDrpAqOgOS7zQg,2733
@@ -159,6 +159,10 @@ optimum/rbln/transformers/models/qwen2_5_vl/__init__.py,sha256=rAW3DKQUzGL6EMwa5
 optimum/rbln/transformers/models/qwen2_5_vl/configuration_qwen2_5_vl.py,sha256=lyT0U3Em7l3Y3ZETVEniDbEliVI6xGbYO0LSeqXSJNA,4763
 optimum/rbln/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py,sha256=APlEF_YAvhJdsTnYcmZuWe47lUZ_3fozhQA_bmCIYV4,26638
 optimum/rbln/transformers/models/qwen2_5_vl/qwen2_5_vl_architecture.py,sha256=oU4MyNeDHzqD3dl1DgwrMev07yvMFhl_hXvV6tRdXCo,7422
+optimum/rbln/transformers/models/qwen3/__init__.py,sha256=tI4KwvXpD35dUUaa8aLUXpWoU9gJGcmKXeywOlH14ZE,746
+optimum/rbln/transformers/models/qwen3/configuration_qwen3.py,sha256=Si_O9ir0a8nzUVeV8ClKqNvFy8hzP9JW21YBim_gEmc,2085
+optimum/rbln/transformers/models/qwen3/modeling_qwen3.py,sha256=0hn_mWnZZqqwbUYnTvvMutNqec-HevCuCwZ8yxFSeuk,15716
+optimum/rbln/transformers/models/qwen3/qwen3_architecture.py,sha256=4756suDvAkkoXbQU0l6ACOcIlDjSj5gxjJsOFfyT3GU,10767
 optimum/rbln/transformers/models/resnet/__init__.py,sha256=0QqtEQF1IMYgEmmfXMGarCDS8kJB5tzODfwTEzDVZRg,837
 optimum/rbln/transformers/models/resnet/configuration_resnet.py,sha256=aOHNDpSi630H3LhDx_8aWh6cSR_zA7rgoMdp6MiiPl8,983
 optimum/rbln/transformers/models/resnet/modeling_resnet.py,sha256=6xKvD8HQCL-e-NtfvMrlL1fBWJBQtR_GK9zKTg2OJPQ,1021
@@ -167,7 +171,7 @@ optimum/rbln/transformers/models/roberta/configuration_roberta.py,sha256=6KhO-xB
 optimum/rbln/transformers/models/roberta/modeling_roberta.py,sha256=74Pswb5JJNtctvrQHlo2zYocKZN0npWhjAaKMUDVBUU,1535
 optimum/rbln/transformers/models/seq2seq/__init__.py,sha256=HiSyWFcKeZ8okfo-s-_Mf_upyvAoZwraUIJyGNLNurY,714
 optimum/rbln/transformers/models/seq2seq/configuration_seq2seq.py,sha256=lOLCGd9sv4N2fYsJmbf4eFPMpI5NFtA656qCEjpfulE,2651
-optimum/rbln/transformers/models/seq2seq/modeling_seq2seq.py,sha256=a9VNsbUtFEOTnvs6OIEH0-3O2OQC7CIZrJStG5ub2bA,17163
+optimum/rbln/transformers/models/seq2seq/modeling_seq2seq.py,sha256=WApUSR0KFr6a0C3Yh5IAxH2-8w8HT6QC5JlzVZqoJeE,17253
 optimum/rbln/transformers/models/seq2seq/seq2seq_architecture.py,sha256=4MupGjhe_DpBLl43VVYY72WWiVdAr0mRPp8cHalQh2w,20028
 optimum/rbln/transformers/models/siglip/__init__.py,sha256=39MdhvWLZU0_9julQtJYVEiWI4csPrryS9krMauEA3s,730
 optimum/rbln/transformers/models/siglip/configuration_siglip.py,sha256=m1h8iDx_X9VmHdJi0sc1a2KsAO3OnpMb4cd9jW2Ic-U,3031
@@ -178,7 +182,7 @@ optimum/rbln/transformers/models/t5/modeling_t5.py,sha256=pdAWBLVknTzbma0Ij-VQ2Q
 optimum/rbln/transformers/models/t5/t5_architecture.py,sha256=X_9X4QRhkiiMrwFHv3mzER3yGmF9oQ2U-HdH6jbwVmw,9824
 optimum/rbln/transformers/models/time_series_transformer/__init__.py,sha256=xJaFWQawlwtv4H5tVFcY1pxLYzjHtMAlLq6nXysdkN8,1243
 optimum/rbln/transformers/models/time_series_transformer/configuration_time_series_transformer.py,sha256=9tomhFoGtLoWyrsAwbS6ArHZ9TcJy2M3Bm2PqR457_Y,1638
-optimum/rbln/transformers/models/time_series_transformer/modeling_time_series_transformer.py,sha256=_03leZBn9OsmnW_H-D40s7CR4Pwqkh7u0cQW3GZqDOE,17242
+optimum/rbln/transformers/models/time_series_transformer/modeling_time_series_transformer.py,sha256=XCLw_X8dXXus6CAQPozy_VVUV9vaFlyupVNMiN4wWuk,17332
 optimum/rbln/transformers/models/time_series_transformer/time_series_transformers_architecture.py,sha256=XJDjQGbWXUq4ZimNojlcbm3mTDpxUMCl6tkFSzfYFl4,13769
 optimum/rbln/transformers/models/vit/__init__.py,sha256=CrrkHehfCe3U-_rUS00aMBY7Tncdeh43sNUgVI9Dt_g,807
 optimum/rbln/transformers/models/vit/configuration_vit.py,sha256=x98CxKR1cpKAG7Eh43uuPeGeGn4gS3HcKLPoDL3SWJo,994
@@ -189,7 +193,7 @@ optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py,sha256=vp5l0zEjeE
 optimum/rbln/transformers/models/whisper/__init__.py,sha256=ErquiUlYycSYPsDcq9IwwmbZXoYLn1MVZ8VikWY5gQo,792
 optimum/rbln/transformers/models/whisper/configuration_whisper.py,sha256=KjM11IRvlwXR1hs811onLi8dxm1-N59ajxe9X7wvfr0,2900
 optimum/rbln/transformers/models/whisper/generation_whisper.py,sha256=GIHTca3b1VtW81kp7BzKQ7f77c2t9OsEsbZetripgDo,4582
-optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=MAcc9fYjpqywD5IS7y5ayXWbeJTij0XQoag4wakj2xg,18058
+optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=qGv9GavffGe_EMK5z5PffbaiYxJc6QdljxRaq6SuMG4,18148
 optimum/rbln/transformers/models/whisper/whisper_architecture.py,sha256=fKUbAMIl20o6EBMVcLg9TDSsJ1FDp8NKcl4jT9RWCEM,13981
 optimum/rbln/transformers/models/xlm_roberta/__init__.py,sha256=S-ksBzS3Jg4lJlBjagU1qrmKsJ9xOBXKJU3OrGBF_po,1008
 optimum/rbln/transformers/models/xlm_roberta/configuration_xlm_roberta.py,sha256=wHRpGTXL9khYqSkKL1IgA7__6_lt9QpOz9tHumjK7fo,1260
@@ -202,10 +206,10 @@ optimum/rbln/utils/hub.py,sha256=Z_R9Ic9VAew8bUmlaAlxZf5JGMDBivHvvFRI557pILY,419
 optimum/rbln/utils/import_utils.py,sha256=fpOERIIxXm-cDYGn1NN6c7aWDPQYVitPQW2MiyZ9NEY,5471
 optimum/rbln/utils/logging.py,sha256=VKKBmlQSdg6iZCGmAXaWYiW67K84jyp1QJhLQSSjPPE,3453
 optimum/rbln/utils/model_utils.py,sha256=4k5879Kh75m3x_vS4-qOGfqsOiAvc2kdNFFfvsFvz3k,1748
-optimum/rbln/utils/runtime_utils.py,sha256=D9PS8hfH1NBf8yH8cAu-XfdC9fxKzPbt4LFBVpADbbs,7180
+optimum/rbln/utils/runtime_utils.py,sha256=nIJioiN16nAyAzoArKjsy5ocLUsrr0UEy4f3LNT82SA,7961
 optimum/rbln/utils/save_utils.py,sha256=hG5uOtYmecSXZuGTvCXsTM-SiyZpr5q3InUGCCq_jzQ,3619
 optimum/rbln/utils/submodule.py,sha256=w5mgPgncI740gVKMu3S-69DGNdUSI0bTZxegQGcZ98Y,5011
-optimum_rbln-0.8.2a0.dist-info/METADATA,sha256=dHMIEdFF_IuTWww99Iypz6HQKVDDk___EVJ8cK77eG0,5299
-optimum_rbln-0.8.2a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-optimum_rbln-0.8.2a0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-optimum_rbln-0.8.2a0.dist-info/RECORD,,
+optimum_rbln-0.8.2a2.dist-info/METADATA,sha256=ncxtO1-Q-UrBAF4eatV3ZKNIpdpj5nwFNmHdxvJ0c3U,5299
+optimum_rbln-0.8.2a2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+optimum_rbln-0.8.2a2.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+optimum_rbln-0.8.2a2.dist-info/RECORD,,