optimum-rbln 0.8.1a7__py3-none-any.whl → 0.8.1rc1__py3-none-any.whl

optimum/rbln/__version__.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.8.1a7'
-__version_tuple__ = version_tuple = (0, 8, 1, 'a7')
+__version__ = version = '0.8.1rc1'
+__version_tuple__ = version_tuple = (0, 8, 1, 'rc1')

optimum/rbln/diffusers/configurations/models/__init__.py

@@ -1,8 +1,8 @@
 from .configuration_autoencoder_kl import RBLNAutoencoderKLConfig
 from .configuration_autoencoder_kl_cosmos import RBLNAutoencoderKLCosmosConfig
 from .configuration_controlnet import RBLNControlNetModelConfig
-from .configuration_cosmos_transformer import RBLNCosmosTransformer3DModelConfig
 from .configuration_prior_transformer import RBLNPriorTransformerConfig
+from .configuration_transformer_cosmos import RBLNCosmosTransformer3DModelConfig
 from .configuration_transformer_sd3 import RBLNSD3Transformer2DModelConfig
 from .configuration_unet_2d_condition import RBLNUNet2DConditionModelConfig
 from .configuration_vq_model import RBLNVQModelConfig

optimum/rbln/diffusers/configurations/models/configuration_autoencoder_kl_cosmos.py

@@ -22,6 +22,8 @@ logger = get_logger(__name__)
 
 
 class RBLNAutoencoderKLCosmosConfig(RBLNModelConfig):
+    """Configuration class for RBLN Cosmos Variational Autoencoder (VAE) models."""
+
     def __init__(
         self,
         batch_size: Optional[int] = None,
@@ -48,7 +50,7 @@ class RBLNAutoencoderKLCosmosConfig(RBLNModelConfig):
                 Determines how much shorter the latent representations are compared to the original videos.
             vae_scale_factor_spatial (Optional[int]): The scaling factor between pixel space and latent space.
                 Determines how much smaller the latent representations are compared to the original videos.
-            use_slicing (Optional[Bool]): Enable sliced VAE encoding and decoding.
+            use_slicing (Optional[bool]): Enable sliced VAE encoding and decoding.
                 If True, the VAE will split the input tensor in slices to compute encoding or decoding in several steps.
             **kwargs: Additional arguments passed to the parent RBLNModelConfig.
 

optimum/rbln/diffusers/configurations/models/{configuration_cosmos_transformer.py → configuration_transformer_cosmos.py}

@@ -18,6 +18,8 @@ from ....configuration_utils import RBLNModelConfig
 
 
 class RBLNCosmosTransformer3DModelConfig(RBLNModelConfig):
+    """Configuration class for RBLN Cosmos Transformer models."""
+
     def __init__(
         self,
         batch_size: Optional[int] = None,

optimum/rbln/diffusers/configurations/pipelines/configuration_cosmos.py

@@ -24,7 +24,7 @@ from ..models import RBLNAutoencoderKLCosmosConfig, RBLNCosmosTransformer3DModel
 logger = get_logger(__name__)
 
 
-class _RBLNCosmosPipelineBaseConfig(RBLNModelConfig):
+class RBLNCosmosPipelineBaseConfig(RBLNModelConfig):
     submodules = ["text_encoder", "transformer", "vae", "safety_checker"]
     _vae_uses_encoder = False
 
@@ -47,7 +47,7 @@ class _RBLNCosmosPipelineBaseConfig(RBLNModelConfig):
         Args:
             text_encoder (Optional[RBLNT5EncoderModelConfig]): Configuration for the text encoder component.
                 Initialized as RBLNT5EncoderModelConfig if not provided.
-            transformer (Optional[RBLNCosmosTransformer3DModelConfig]): Configuration for the UNet model component.
+            transformer (Optional[RBLNCosmosTransformer3DModelConfig]): Configuration for the Transformer model component.
                 Initialized as RBLNCosmosTransformer3DModelConfig if not provided.
             vae (Optional[RBLNAutoencoderKLCosmosConfig]): Configuration for the VAE model component.
                 Initialized as RBLNAutoencoderKLCosmosConfig if not provided.
@@ -102,9 +102,13 @@ class _RBLNCosmosPipelineBaseConfig(RBLNModelConfig):
         return self.text_encoder.max_seq_len
 
 
-class RBLNCosmosTextToWorldPipelineConfig(_RBLNCosmosPipelineBaseConfig):
+class RBLNCosmosTextToWorldPipelineConfig(RBLNCosmosPipelineBaseConfig):
+    """Config for Cosmos Text2World Pipeline"""
+
     _vae_uses_encoder = False
 
 
-class RBLNCosmosVideoToWorldPipelineConfig(_RBLNCosmosPipelineBaseConfig):
+class RBLNCosmosVideoToWorldPipelineConfig(RBLNCosmosPipelineBaseConfig):
+    """Config for Cosmos Video2World Pipeline"""
+
     _vae_uses_encoder = True

optimum/rbln/diffusers/pipelines/cosmos/cosmos_guardrail.py

@@ -130,7 +130,7 @@ class RBLNSigLIPEncoder(SigLIPEncoder):
                 self.checkpoint_dir,
                 rbln_device=rbln_config.siglip_encoder.device,
                 rbln_create_runtimes=rbln_config.siglip_encoder.create_runtimes,
-                rbln_activate_profiler=rbln_config.aegis.activate_profiler,
+                rbln_activate_profiler=rbln_config.siglip_encoder.activate_profiler,
             )
         else:
             super().__init__(model_name, checkpoint_id)
@@ -143,6 +143,7 @@ class RBLNSigLIPEncoder(SigLIPEncoder):
                 rbln_npu=rbln_config.siglip_encoder.npu,
                 rbln_create_runtimes=rbln_config.siglip_encoder.create_runtimes,
                 rbln_activate_profiler=rbln_config.siglip_encoder.activate_profiler,
+                rbln_optimize_host_memory=rbln_config.siglip_encoder.optimize_host_memory,
             )
         self.rbln_config = rbln_config
 
@@ -353,6 +354,7 @@ class RBLNAegis(Aegis):
                 rbln_create_runtimes=rbln_config.aegis.create_runtimes,
                 rbln_npu=rbln_config.aegis.npu,
                 rbln_activate_profiler=rbln_config.aegis.activate_profiler,
+                rbln_optimize_host_memory=rbln_config.aegis.optimize_host_memory,
             )
 
         self.rbln_config = rbln_config

optimum/rbln/diffusers/pipelines/cosmos/pipeline_cosmos_text2world.py

@@ -35,7 +35,7 @@ class RBLNCosmosTextToWorldPipeline(RBLNDiffusionMixin, CosmosTextToWorldPipelin
     RBLN-accelerated implementation of Cosmos Text to World pipeline for text-to-video generation.
 
     This pipeline compiles Cosmos Text to World models to run efficiently on RBLN NPUs, enabling high-performance
-    inference for generating images with distinctive artistic style and enhanced visual quality.
+    inference for generating videos with distinctive artistic style and enhanced visual quality.
     """
 
     original_class = CosmosTextToWorldPipeline

optimum/rbln/diffusers/pipelines/cosmos/pipeline_cosmos_video2world.py

@@ -35,12 +35,12 @@ class RBLNCosmosVideoToWorldPipeline(RBLNDiffusionMixin, CosmosVideoToWorldPipel
     RBLN-accelerated implementation of Cosmos Video to World pipeline for video-to-video generation.
 
     This pipeline compiles Cosmos Video to World models to run efficiently on RBLN NPUs, enabling high-performance
-    inference for generating images with distinctive artistic style and enhanced visual quality.
+    inference for generating videos with distinctive artistic style and enhanced visual quality.
     """
 
     original_class = CosmosVideoToWorldPipeline
     _submodules = ["text_encoder", "transformer", "vae"]
-    _optional_components = ["safety_checker"]
+    _optional_submodules = ["safety_checker"]
 
     def __init__(
         self,

{optimum_rbln-0.8.1a7.dist-info → optimum_rbln-0.8.1rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimum-rbln
-Version: 0.8.1a7
+Version: 0.8.1rc1
 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.1a7.dist-info → optimum_rbln-0.8.1rc1.dist-info}/RECORD

@@ -1,23 +1,23 @@
 optimum/rbln/__init__.py,sha256=MZCYmY4Y_Zfk0TGo3xK52osHDLZHz4cSdduXZt6RfSI,15316
-optimum/rbln/__version__.py,sha256=bZCt4qfFUJwL5Nkf3QBkWKhjvo10DoUgDk-05HbCr4Q,519
+optimum/rbln/__version__.py,sha256=_HBfPPqmrksIsmr93RnGS_EXQHyrC_e0NtnXkbI_leI,521
 optimum/rbln/configuration_utils.py,sha256=o5oer7fBdE-MHLGNXoP35FjmuQbMmjEIDv0QE_k3kpo,32336
 optimum/rbln/modeling.py,sha256=bsvK6GQtoH9vx72Ea59kvv61jguOk9XDTzVjsY1ugkk,14248
 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/configurations/__init__.py,sha256=vMRnPY4s-Uju43xP038D2EA18X_mhy2YfsZVpSU-VoA,1322
-optimum/rbln/diffusers/configurations/models/__init__.py,sha256=-4oanQrr9IVznY-ly9ivEnR_zFlfMxzLV6ixLH6TT3w,567
+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=brjtidoogeCc4fYuJ7GxWkJylSu_QGPwJT8l_FQldBI,4060
+optimum/rbln/diffusers/configurations/models/configuration_autoencoder_kl_cosmos.py,sha256=0BPZ4Tcz72j0a1KPdfpvk5G511wWNvS4MDJKbHAPzhA,4145
 optimum/rbln/diffusers/configurations/models/configuration_controlnet.py,sha256=VDO_YFS_QhcHhuRIXQL53JZXEO27yoKHtecq5hd2la8,2637
-optimum/rbln/diffusers/configurations/models/configuration_cosmos_transformer.py,sha256=MnRjn6yI7ST6QFU2MsZqdE2I_FE6vJOOWw4FTTi3KHo,3033
 optimum/rbln/diffusers/configurations/models/configuration_prior_transformer.py,sha256=vE8RsXc27Z4-9k0KEM_vP7AWd5UUYvDgfX1g6nUrPp4,2224
+optimum/rbln/diffusers/configurations/models/configuration_transformer_cosmos.py,sha256=WOsC3sufsbwEEtsx0q8KqXUQE2VAFFCKRC7f2H4fmhg,3100
 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
 optimum/rbln/diffusers/configurations/pipelines/__init__.py,sha256=RfJXQiYvgGc3Rp7JYk5s0AQd0XB5JCAb37_riGWQAYg,1268
 optimum/rbln/diffusers/configurations/pipelines/configuration_controlnet.py,sha256=nTtr2vqyr3zNSJXI0kiTAhOnVNhA-cVyaSnKOwBBZIo,14215
-optimum/rbln/diffusers/configurations/pipelines/configuration_cosmos.py,sha256=iECUFpM453j3QX54v-idh-_wi1_GfMvtiaawfUOUxl4,4485
+optimum/rbln/diffusers/configurations/pipelines/configuration_cosmos.py,sha256=8GK9pAGzhtmQN2DQjgQZJo9S7y6dDko-Ii-3CagFxo4,4588
 optimum/rbln/diffusers/configurations/pipelines/configuration_kandinsky2_2.py,sha256=1ve6o4OEpjPzTXWHXy_T5MAI0V-F08PMv2W6nBFfeKU,16386
 optimum/rbln/diffusers/configurations/pipelines/configuration_stable_diffusion.py,sha256=kR8dV_RsmoDxhK5bAfv3PbtS5LpN5g-O-snAX1sP6Fo,6591
 optimum/rbln/diffusers/configurations/pipelines/configuration_stable_diffusion_3.py,sha256=f2VOwvCd-9kDnUpwhb0LaMWgfwdmBzUKMpmCdhUv2sc,7923
@@ -44,9 +44,9 @@ optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py,sha256=
 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=q2PVTaMDd_IEl_VHmGSCrDJpLamZ2dgmPiFb87TTGhw,17906
-optimum/rbln/diffusers/pipelines/cosmos/pipeline_cosmos_text2world.py,sha256=m1RQr5C1G5GUI1bXI7BvZN_WWl75LvYFyC2ApoEaUAI,3910
-optimum/rbln/diffusers/pipelines/cosmos/pipeline_cosmos_video2world.py,sha256=icJXQbzCZygDuDs_PDJtq7qQr9uX3q-ljJOT6sXB8JQ,3917
+optimum/rbln/diffusers/pipelines/cosmos/cosmos_guardrail.py,sha256=lzT6j0RifZfIktETC5gd8GAKPHwekeb3JH56aPChOtI,18088
+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
 optimum/rbln/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py,sha256=2vgZZt0JrZlbHbUlBHrT0zKvYhuX1a4vwy3mxTPHisM,1335
 optimum/rbln/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py,sha256=cuN7w5LsnkSy0WUaCBPxVmYOJiBMihdh-KM7iV0tHL0,8149
@@ -205,7 +205,7 @@ optimum/rbln/utils/model_utils.py,sha256=4k5879Kh75m3x_vS4-qOGfqsOiAvc2kdNFFfvsF
 optimum/rbln/utils/runtime_utils.py,sha256=D9PS8hfH1NBf8yH8cAu-XfdC9fxKzPbt4LFBVpADbbs,7180
 optimum/rbln/utils/save_utils.py,sha256=hG5uOtYmecSXZuGTvCXsTM-SiyZpr5q3InUGCCq_jzQ,3619
 optimum/rbln/utils/submodule.py,sha256=w5mgPgncI740gVKMu3S-69DGNdUSI0bTZxegQGcZ98Y,5011
-optimum_rbln-0.8.1a7.dist-info/METADATA,sha256=r2_TKXM5kAjO19gni1ztcEFlR4mU1yYj47qdL3rPZgM,5299
-optimum_rbln-0.8.1a7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-optimum_rbln-0.8.1a7.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-optimum_rbln-0.8.1a7.dist-info/RECORD,,
+optimum_rbln-0.8.1rc1.dist-info/METADATA,sha256=52EcXRgXYVTSOWiLL_fQhxzsbZFrESQLFpluOvePehw,5300
+optimum_rbln-0.8.1rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+optimum_rbln-0.8.1rc1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+optimum_rbln-0.8.1rc1.dist-info/RECORD,,