optimum-rbln 0.2.0__py3-none-any.whl → 0.2.1a0__py3-none-any.whl

optimum/rbln/__version__.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.0'
-__version_tuple__ = version_tuple = (0, 2, 0)
+__version__ = version = '0.2.1a0'
+__version_tuple__ = version_tuple = (0, 2, 1)

optimum/rbln/transformers/utils/rbln_quantization.py

@@ -52,7 +52,7 @@ class QuantizationManager:
     ) -> None:
         context_info = f" for {context}" if context else ""
         valid_values_str = ", ".join(valid_values)
-        raise ValueError(f"Invalid {key}: {value}{context_info}. " f"Supported values are: {valid_values_str}")
+        raise ValueError(f"Invalid {key}: {value}{context_info}. Supported values are: {valid_values_str}")
 
     @staticmethod
     def validate_quantization_config(quantize_config: Optional[dict]) -> Optional[dict]:

optimum/rbln/utils/import_utils.py

@@ -132,7 +132,7 @@ def is_rbln_available() -> bool:
 def check_version_compats() -> None:
     warnings.filterwarnings(action="always", category=ImportWarning, module="optimum.rbln")
     my_version = importlib.metadata.version("optimum-rbln")
-    target_version = list(filter(lambda v: Version(my_version) > Version(v), RBLN_VERSION_COMPATS.keys()))[0]
+    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)

optimum/rbln/utils/logging.py

@@ -70,7 +70,7 @@ def _get_default_logging_level():
         else:
             logging.getLogger().warning(
                 f"Unknown option OPTIMUM_RBLN_VERBOSE={env_level_str}, "
-                f"has to be one of: { ', '.join(log_levels.keys()) }"
+                f"has to be one of: {', '.join(log_levels.keys())}"
             )
     return _default_log_level
 

{optimum_rbln-0.2.0.dist-info → optimum_rbln-0.2.1a0.dist-info}/METADATA

@@ -1,9 +1,10 @@
 Metadata-Version: 2.3
 Name: optimum-rbln
-Version: 0.2.0
+Version: 0.2.1a0
 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
+Project-URL: Repository, https://github.com/rebellions-sw/optimum-rbln
 Author-email: "Rebellions Inc." <support@rebellions.ai>
 License: Apache
 Keywords: atom,diffusers,inference,rbln,rebel,transformers
@@ -22,13 +23,11 @@ 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
+Requires-Dist: transformers==4.48.2
 Description-Content-Type: text/markdown
 
 
@@ -39,7 +38,9 @@ Description-Content-Type: text/markdown
 <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)
+[![Documentation](https://img.shields.io/badge/docs-available-brightgreen)](https://docs.rbln.ai/software/optimum/optimum_rbln.html)
+[![Contributor Covenant](https://img.shields.io/badge/Contributor%20Covenant-2.1-4baaaa.svg)](CODE_OF_CONDUCT.md)
+
 
 </div>
 
@@ -95,6 +96,9 @@ Check out [the documentation of Optimum RBLN](https://docs.rbln.ai/software/opti
 To install the latest release of this package:
 ```bash
 pip install optimum-rbln
+
+# CPU-only installation (recommended if you don't plan to use CUDA-enabled PyTorch)
+pip install optimum-rbln --extra-index-url https://download.pytorch.org/whl/cpu
 ```
 
 ### Install from source

{optimum_rbln-0.2.0.dist-info → optimum_rbln-0.2.1a0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 optimum/rbln/__init__.py,sha256=A0ydlEd6-L_ol6wfIFaKl3K93O1j6j9lCO9edPs7MJE,6430
-optimum/rbln/__version__.py,sha256=H-qsvrxCpdhaQzyddR-yajEqI71hPxLa4KxzpP3uS1g,411
+optimum/rbln/__version__.py,sha256=g6hYCmG5a0qiBpcPwa8bCxg_4_YR3PJcAHPGPixF6ho,413
 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
@@ -98,17 +98,17 @@ optimum/rbln/transformers/models/whisper/whisper_architecture.py,sha256=0JKv4GCD
 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/transformers/utils/rbln_quantization.py,sha256=7islo6dn686UKyGrofvl1FKKibkmzHIxk7ZUGNF1j1E,7802
 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/import_utils.py,sha256=ZMRbqZ_UOt7kpTc3LMh8J0CEqgWPV0iz6GZP2nCF69o,4453
+optimum/rbln/utils/logging.py,sha256=JI9rTxtC_7mJ8dVOGSF7Mmb8TO0YjzcRV9W_4gKLABQ,3847
 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.2.1a0.dist-info/METADATA,sha256=efBIm4ovTfE7kHguvu3rndjxjI88UwkFbcMEE38KCN8,5119
+optimum_rbln-0.2.1a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+optimum_rbln-0.2.1a0.dist-info/licenses/LICENSE,sha256=q7yy_9ECqQIxXbh2yiG4Hjalfp0e2-2Pbdc7psKUVs8,18079
+optimum_rbln-0.2.1a0.dist-info/RECORD,,