optimum-rbln 0.7.2rc0__tar.gz → 0.7.2rc1__tar.gz

{optimum_rbln-0.7.2rc0 → optimum_rbln-0.7.2rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimum-rbln
-Version: 0.7.2rc0
+Version: 0.7.2rc1
 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

{optimum_rbln-0.7.2rc0 → optimum_rbln-0.7.2rc1}/src/optimum/rbln/__version__.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.7.2rc0'
+__version__ = version = '0.7.2rc1'
 __version_tuple__ = version_tuple = (0, 7, 2)

{optimum_rbln-0.7.2rc0 → optimum_rbln-0.7.2rc1}/src/optimum/rbln/diffusers/modeling_diffusers.py

@@ -414,9 +414,16 @@ class RBLNDiffusionMixin:
         if rbln_config.get("optimize_host_memory") is False:
             # Keep compiled_model objs to further analysis. -> TODO: remove soon...
             model.compiled_models = []
-            for name in cls._submodules:
-                submodule = getattr(model, name)
-                model.compiled_models.extend(submodule.compiled_models)
+            if model._load_connected_pipes:
+                for name in cls._submodules:
+                    connected_pipe = getattr(model, name)
+                    for submodule_name in connected_pipe.__class__._submodules:
+                        submodule = getattr(connected_pipe, submodule_name)
+                        model.compiled_models.extend(submodule.compiled_models)
+            else:
+                for name in cls._submodules:
+                    submodule = getattr(model, name)
+                    model.compiled_models.extend(submodule.compiled_models)
 
         return model