optimum-rbln 0.8.0.post1__tar.gz → 0.8.1__tar.gz

optimum_rbln-0.8.1/.github/scripts/validate_docstrings.py

@@ -0,0 +1,161 @@
+#!/usr/bin/env python3
+"""
+Simple and reliable docstring validator for CI/CD.
+Focuses on catching issues that would break mkdocstrings parsing.
+"""
+
+import subprocess
+import sys
+import tempfile
+from pathlib import Path
+
+
+def infer_module_name(file_path: Path) -> str:
+    """
+    Infer module name from file path.
+
+    Example:
+        src/optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py
+        -> optimum.rbln.transformers.models.decoderonly.modeling_decoderonly
+    """
+    if not any(keyword in file_path.name for keyword in ["configuration", "modeling"]):
+        return None
+
+    parts = file_path.parts
+
+    # Find 'src' directory index
+    try:
+        src_index = parts.index("src")
+    except ValueError:
+        # If no 'src' directory found, try common patterns
+        for common_src in ["lib", "package", file_path.parts[0]]:
+            if common_src in parts:
+                src_index = parts.index(common_src)
+                break
+        else:
+            # Fallback: use the first directory
+            src_index = 0
+
+    # Extract module path parts (everything after src, excluding .py)
+    module_parts = parts[src_index + 1 :]
+
+    # Remove .py extension from the last part
+    if module_parts and module_parts[-1].endswith(".py"):
+        module_parts = module_parts[:-1] + (module_parts[-1][:-3],)
+
+    # Join with dots
+    module_name = ".".join(module_parts)
+    return module_name
+
+
+def test_mkdocstrings_parsing(file_path: Path, module_name: str = None) -> bool:
+    """Test if mkdocstrings can parse the module documentation."""
+    try:
+        if module_name:
+            # Test with specific module name
+            test_content = f"""# Test Documentation
+
+:::{module_name}
+    options:
+        show_source: false
+        filters: ["!^_"]
+"""
+        else:
+            # Auto-infer module name from file path
+            inferred_module = infer_module_name(file_path)
+            if inferred_module is None:
+                print(
+                    "⚠️ No module name provided, and the file is not a configuration or modeling file, skipping mkdocstrings test"
+                )
+                return True
+            print(f"ℹ️  Auto-inferred module name: {inferred_module}")
+            test_content = f"""# Test Documentation
+
+:::{inferred_module}
+    options:
+        show_source: false
+        filters: ["!^_"]
+"""
+            module_name = inferred_module
+
+        config = f"""site_name: Test
+plugins:
+  - mkdocstrings:
+      handlers:
+        python:
+          paths: ["{Path(__file__).parent.parent.parent / "src"}"]
+          options:
+            show_source: false
+            filters: ["!^_"]
+"""
+
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            tmp_path = Path(tmp_dir)
+            (tmp_path / "mkdocs.yml").write_text(config)
+            docs_dir = tmp_path / "docs"
+            docs_dir.mkdir()
+            (docs_dir / "index.md").write_text(test_content)
+
+            result = subprocess.run(
+                ["mkdocs", "build", "--strict"], cwd=tmp_path, capture_output=True, text=True, timeout=10
+            )
+
+            if result.returncode != 0:
+                print(f"❌ mkdocstrings parsing failed for {file_path}:")
+                print(f"   Module: {module_name}")
+                # Show relevant errors and warnings
+                lines = result.stderr.split("\n")
+                relevant = [
+                    l for l in lines if any(keyword in l for keyword in ["griffe", "WARNING", "ERROR", "CRITICAL"])
+                ]
+                for line in relevant[:5]:  # Show first 5 relevant lines
+                    if line.strip():
+                        print(f"    {line.strip()}")
+                return False
+
+            return True
+
+    except subprocess.TimeoutExpired:
+        print(f"⚠️  Timeout testing {file_path} (non-fatal)")
+        return True  # Don't fail on timeout
+    except Exception as e:
+        print(f"⚠️  Could not test mkdocstrings for {file_path}: {e} (non-fatal)")
+        return True  # Don't fail on test errors
+
+
+def main():
+    """Main entry point."""
+    if len(sys.argv) < 2:
+        print("Usage: python validate_docstrings.py <python_file> [module_name]")
+        print()
+        print("Examples:")
+        print("  python validate_docstrings.py src/my_module.py")
+        print("  python validate_docstrings.py src/my_module.py my.module.name")
+        print()
+        print("Note: If module_name is not provided, it will be auto-inferred from file path")
+        sys.exit(1)
+
+    file_path = Path(sys.argv[1])
+    module_name = sys.argv[2] if len(sys.argv) > 2 else None
+
+    if not file_path.exists():
+        print(f"❌ File not found: {file_path}")
+        sys.exit(1)
+
+    print(f"🔍 Testing mkdocstrings parsing: {file_path}")
+    if module_name:
+        print(f"📦 Module: {module_name}")
+
+    # Test mkdocstrings compatibility
+    mkdocs_ok = test_mkdocstrings_parsing(file_path, module_name)
+
+    if mkdocs_ok:
+        print("✅ mkdocstrings parsing test passed!")
+        sys.exit(0)
+    else:
+        print("❌ mkdocstrings parsing test failed!")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

optimum_rbln-0.8.1/.github/version.yaml

@@ -0,0 +1 @@
+rebel_compiler_version: 0.8.1.dev142+gab6ad3c7

{optimum_rbln-0.8.0.post1 → optimum_rbln-0.8.1}/.github/workflows/check_code_quality.yml

@@ -1,19 +1,7 @@
 name: check_code_quality
 
 on:
-  push:
-    branches: [main]
-    paths:
-      - "src/**/*.py"
-      - "tests/**/*.py"
-      - "examples/**/*.py"
-
-  pull_request:
-    branches: [main]
-    paths:
-      - "src/**/*.py"
-      - "tests/**/*.py"
-      - "examples/**/*.py"
+  workflow_call:
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}

{optimum_rbln-0.8.0.post1 → optimum_rbln-0.8.1}/.github/workflows/rbln_scheduled_test.yaml

@@ -10,6 +10,14 @@ env:
   HF_AUTH_TOKEN: ${{ secrets.HF_AUTH_TOKEN }}
 
 jobs:
+  check-code-quality:
+    uses: ./.github/workflows/check_code_quality.yml
+    
+  test-docstrings:
+    uses: ./.github/workflows/test-docstrings.yml
+    with:
+      test_all_files: true
+
   load-version:
     runs-on: rebel-k8s-runner
     outputs:

{optimum_rbln-0.8.0.post1 → optimum_rbln-0.8.1}/.github/workflows/rbln_trigger_on_pr.yaml

@@ -24,9 +24,19 @@ jobs:
             echo "No [skip ci] found, continuing with CI"
           fi
 
+  check-code-quality:
+    needs: check-skip-ci
+    if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' }}
+    uses: ./.github/workflows/check_code_quality.yml
+    
+  test-docstrings:
+    needs: check-skip-ci
+    if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' }}
+    uses: ./.github/workflows/test-docstrings.yml
+
   load-version:
     runs-on: rebel-k8s-runner
-    needs: check-skip-ci
+    needs: [check-skip-ci, check-code-quality, test-docstrings]
     if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' }}
     outputs:
       compiler_version: ${{ steps.get_version.outputs.compiler_version }}
@@ -41,7 +51,7 @@ jobs:
           echo "compiler_version=$VERSION" >> $GITHUB_OUTPUT
 
   check-compiler:
-    needs: [check-skip-ci, load-version]
+    needs: [check-skip-ci, check-code-quality, test-docstrings, load-version]
     if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' }}
     uses: ./.github/workflows/rbln_check_compiler.yaml
     with:
@@ -50,7 +60,7 @@ jobs:
 
   check-team-member:
     runs-on: rebel-k8s-runner
-    needs: [check-skip-ci, check-compiler]
+    needs: [check-skip-ci, check-code-quality, test-docstrings, check-compiler]
     if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' && needs.check-compiler.outputs.is_deploy_version == 'true' }}
     outputs:
       is_team_member: ${{ steps.check_member.outputs.IS_TEAM_MEMBER }}
@@ -75,7 +85,7 @@ jobs:
           fi
 
   optimum-rbln-pytest:
-    needs: [check-skip-ci, check-compiler, check-team-member]
+    needs: [check-skip-ci, check-code-quality, test-docstrings, check-compiler, check-team-member]
     if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' && needs.check-team-member.outputs.is_team_member == 'true' }}
     uses: ./.github/workflows/rbln_optimum_pytest.yaml
     with:
@@ -85,7 +95,7 @@ jobs:
     secrets: inherit
 
   optimum-rbln-inference-test:
-    needs: [check-skip-ci, check-compiler, check-team-member]
+    needs: [check-skip-ci, check-code-quality, test-docstrings, check-compiler, check-team-member]
     if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' && needs.check-team-member.outputs.is_team_member == 'true' }}
     uses: ./.github/workflows/rbln_optimum_inference_test.yaml
     with:

optimum_rbln-0.8.1/.github/workflows/test-docstrings.yml

@@ -0,0 +1,137 @@
+name: Test Docstring Parsing
+
+on:
+  workflow_call:
+    inputs:
+      test_all_files:
+        description: "Test all files in src/ instead of just changed files"
+        required: false
+        type: boolean
+        default: false
+
+jobs:
+  test-docstrings:
+    runs-on: ubuntu-latest-rbln
+    
+    steps:
+    - uses: actions/checkout@v4
+      with:
+        fetch-depth: 0  # Get full git history for file diff
+    
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.9'
+    
+    - name: Install dependencies
+      run: |
+        pip install mkdocs mkdocs-material mkdocstrings mkdocstrings-python
+    
+    - name: Get Python files to test
+      id: changed-files
+      run: |
+        if [ "${{ inputs.test_all_files }}" = "true" ]; then
+          # For scheduled test, get all Python files in src/
+          ALL_FILES=$(find src/ -name '*.py' -type f || true)
+          echo "Testing all Python files in src/:"
+          echo "$ALL_FILES"
+          
+          if [ -n "$ALL_FILES" ]; then
+            FILES_JSON=$(echo "$ALL_FILES" | jq -R -s -c 'split("\n")[:-1]')
+            echo "files=$FILES_JSON" >> $GITHUB_OUTPUT
+            echo "has_changes=true" >> $GITHUB_OUTPUT
+          else
+            echo "has_changes=false" >> $GITHUB_OUTPUT
+          fi
+        else
+          # For PR, compare with base branch
+          if [ "${{ github.event_name }}" = "pull_request" ]; then
+            CHANGED_FILES=$(git diff --name-only origin/${{ github.base_ref }}...HEAD | grep '\.py$' | grep 'src/' || true)
+          else
+            # For push, compare with previous commit
+            CHANGED_FILES=$(git diff --name-only HEAD~1 HEAD | grep '\.py$' | grep 'src/' || true)
+          fi
+          
+          echo "Changed Python files:"
+          echo "$CHANGED_FILES"
+          
+          # Convert to JSON array for matrix
+          if [ -n "$CHANGED_FILES" ]; then
+            FILES_JSON=$(echo "$CHANGED_FILES" | jq -R -s -c 'split("\n")[:-1]')
+            echo "files=$FILES_JSON" >> $GITHUB_OUTPUT
+            echo "has_changes=true" >> $GITHUB_OUTPUT
+          else
+            echo "has_changes=false" >> $GITHUB_OUTPUT
+          fi
+        fi
+    
+    - name: Test docstring parsing
+      if: steps.changed-files.outputs.has_changes == 'true'
+      run: |
+        FILES='${{ steps.changed-files.outputs.files }}'
+        
+        # Exclude files (hardcoded)
+        EXCLUDE_FILES=(
+          "src/optimum/rbln/transformers/models/llava_next/modeling_llava_next.py"
+          "src/optimum/rbln/transformers/models/t5/modeling_t5.py"
+        )
+        
+        # Filter out excluded files
+        FILTERED_FILES=""
+        echo "$FILES" | jq -r '.[]' | while read -r file; do
+          should_exclude=false
+          for exclude_file in "${EXCLUDE_FILES[@]}"; do
+            if [[ "$file" == "$exclude_file" ]]; then
+              echo "⏭️  Skipping excluded file: $file"
+              should_exclude=true
+              break
+            fi
+          done
+          
+          if [[ "$should_exclude" == "false" ]]; then
+            echo "$file" >> /tmp/filtered_files.txt
+          fi
+        done
+        
+        # Check if we have files to process
+        if [[ ! -f /tmp/filtered_files.txt ]] || [[ ! -s /tmp/filtered_files.txt ]]; then
+          echo "No files to process after excluding files"
+          exit 0
+        fi
+        
+        TOTAL_FILES=$(wc -l < /tmp/filtered_files.txt)
+        echo "Processing $TOTAL_FILES files with 16 parallel processes..."
+        
+        # Function to validate a single file
+        validate_file() {
+          local file=$1
+          echo "Testing: $file"
+          python .github/scripts/validate_docstrings.py "$file"
+          if [ $? -ne 0 ]; then
+            echo "❌ Docstring test failed for $file"
+            return 1
+          fi
+          echo "✅ Docstring test passed for $file"
+          return 0
+        }
+        
+        # Export function for use with xargs
+        export -f validate_file
+        
+        # Process files in parallel with 16 processes
+        cat /tmp/filtered_files.txt | xargs -I {} -P 16 -n 1 bash -c 'validate_file "$@"' _ {}
+        
+        # Check if any process failed
+        if [ $? -ne 0 ]; then
+          echo "❌ Some docstring tests failed"
+          exit 1
+        fi
+        
+        echo "🎉 All docstring tests passed!"
+        
+        # Clean up
+        rm -f /tmp/filtered_files.txt
+    
+    - name: No changes detected
+      if: steps.changed-files.outputs.has_changes == 'false'
+      run: echo "No Python files to test" 

{optimum_rbln-0.8.0.post1 → optimum_rbln-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimum-rbln
-Version: 0.8.0.post1
+Version: 0.8.1
 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
@@ -23,12 +23,12 @@ 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: diffusers==0.34.0
 Requires-Dist: packaging>=24.1
-Requires-Dist: torch==2.6.0
-Requires-Dist: torchaudio<=2.6.0
-Requires-Dist: torchvision<=0.21.0
-Requires-Dist: transformers==4.50.3
+Requires-Dist: torch==2.7.0
+Requires-Dist: torchaudio<=2.7.0
+Requires-Dist: torchvision<=0.22.0
+Requires-Dist: transformers==4.51.3
 Description-Content-Type: text/markdown
 
 

optimum_rbln-0.8.1/examples/advanced/custom_class.py

@@ -0,0 +1,159 @@
+"""
+# Advanced Tutorial: Creating Custom RBLN Classes
+
+This tutorial demonstrates how to create custom RBLN model classes for use with the optimum-rbln framework.
+
+## Requirements for Custom Class Creation:
+
+1. You must create two classes:
+   - A custom model class that inherits from RBLNModel
+   - A custom config class that inherits from RBLNModelConfig
+
+2. The naming convention is critical:
+   - Model class: RBLN<OriginalModelName> (e.g., RBLNResNetModel)
+   - Config class: RBLN<OriginalModelName>Config (e.g., RBLNResNetModelConfig)
+
+3. Your custom class MUST implement the _update_rbln_config method
+   - This method configures input shapes and compilation settings
+
+4. You must register both classes with the auto registration system
+   - This enables the framework to discover and use your custom classes
+
+Important: Custom RBLN classes can only be created for models that exist in
+the transformers or diffusers libraries. You cannot create custom RBLN classes
+for completely custom architectures without a corresponding HuggingFace implementation.
+
+This example demonstrates creating a custom RBLN class for the ResNet model from transformers.
+"""
+
+from typing import TYPE_CHECKING, Optional, Tuple, Union
+
+import torch
+from transformers import ResNetModel  # noqa: F401
+from transformers.models.resnet.modeling_resnet import BaseModelOutputWithPoolingAndNoAttention
+
+from optimum.rbln import RBLNAutoConfig, RBLNAutoModel, RBLNCompileConfig, RBLNModel, RBLNModelConfig
+
+
+if TYPE_CHECKING:
+    from transformers import AutoFeatureExtractor, AutoProcessor, AutoTokenizer, PretrainedConfig, PreTrainedModel
+
+
+# --------------------------------------------------------
+# STEP 1: Create a custom model class that extends RBLNModel
+# --------------------------------------------------------
+# A custom RBLN model class must:
+# - Follow the naming pattern: RBLN<OriginalModelName>
+# - Implement the _update_rbln_config method (required)
+# - Define a proper forward method matching the original model's inputs/outputs
+#
+# The _update_rbln_config method is critical - it sets up the input tensor
+# specifications and other compilation parameters needed for the RBLN compiler.
+#
+# For our ResNet example, we need to:
+# 1. Specify the input tensor shape for images (batch_size, channels, height, width)
+# 2. Set the proper configuration options
+# 3. Return the updated RBLN configuration
+class RBLNResNetModel(RBLNModel):
+    @classmethod
+    def _update_rbln_config(
+        cls,
+        preprocessors: Optional[Union["AutoFeatureExtractor", "AutoProcessor", "AutoTokenizer"]] = None,
+        model: Optional["PreTrainedModel"] = None,
+        model_config: Optional["PretrainedConfig"] = None,
+        rbln_config: Optional["RBLNResNetModelConfig"] = None,
+    ) -> "RBLNResNetModelConfig":
+        # Set image_size if not provided
+        if rbln_config.image_size is None:
+            if rbln_config.image_size is None:
+                rbln_config.image_size = model_config.image_size
+
+            if rbln_config.image_size is None:
+                raise ValueError("`image_size` should be specified!")
+
+        # Define input tensor specification for the compiler
+        # Format: (tensor_name, tensor_shape, dtype)
+        input_info = [
+            (
+                "pixel_values",
+                [rbln_config.batch_size, 3, rbln_config.image_size[0], rbln_config.image_size[1]],
+                "float32",
+            )
+        ]
+
+        # Configure compilation settings
+        rbln_config.set_compile_cfgs([RBLNCompileConfig(input_info=input_info)])
+        return rbln_config
+
+    def forward(self, pixel_values, return_dict: Optional[bool] = None, **kwargs):
+        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+
+        # self.model is a list of rebel.Runtime objects
+        # See https://docs.rbln.ai/software/api/python/python_api.html#rebel.rebel_runtime.Runtime for more details
+        output = self.model[0](pixel_values)
+
+        if not return_dict:
+            return output
+        else:
+            return BaseModelOutputWithPoolingAndNoAttention(last_hidden_state=output[0], pooler_output=output[1])
+
+
+# ----------------------------------------------------------------
+# STEP 2: Create a custom configuration class that extends RBLNModelConfig
+# ----------------------------------------------------------------
+# The configuration class must:
+# - Follow the naming pattern: RBLN<OriginalModelName>Config
+# - Define model-specific parameters needed for compilation
+#
+# For ResNet, we need:
+# - batch_size: Batch size for inference
+# - image_size: Input image dimensions (height, width)
+class RBLNResNetModelConfig(RBLNModelConfig):
+    def __init__(self, batch_size: int = None, image_size: Optional[Tuple[int, int]] = None, **kwargs):
+        super().__init__(**kwargs)
+        self.batch_size = batch_size or 1
+        self.image_size = image_size or (224, 224)
+
+
+# ----------------------------------------------------------------
+# STEP 3: Register both classes with the auto registration system
+# ----------------------------------------------------------------
+# Registration is essential for the RBLN framework to:
+# 1. Discover your custom classes when needed
+# 2. Enable using RBLNAutoModel.from_pretrained() with your model type
+# 3. Connect your configuration class to the model class in the framework
+RBLNAutoModel.register(RBLNResNetModel)
+RBLNAutoConfig.register(RBLNResNetModelConfig)
+
+
+# ----------------------------------------------------------------
+# STEP 4: Usage Example - Creating and using the custom RBLN model
+# ----------------------------------------------------------------
+# Initialize the model from a pretrained HuggingFace model
+# The 'export=True' parameter triggers the model compilation process
+# rbln_image_size and rbln_batch_size are passed to the custom configuration
+my_model = RBLNResNetModel.from_pretrained(
+    "microsoft/resnet-50", export=True, rbln_image_size=(224, 224), rbln_batch_size=1
+)
+
+# Save the compiled model for later use
+my_model.save_pretrained("my_resnet_model_saved")
+
+# Create a random input tensor for demonstration
+# The shape must match what we defined in _update_rbln_config
+random_image_input = torch.randn(1, 3, 224, 224)
+
+# Run inference with the compiled model
+output = my_model(random_image_input)
+
+# Print each key-value pair in the output dictionary
+for key, value in output.items():
+    print(key, value)
+
+
+# ----------------------------------------------------------------
+# STEP 5: Loading a saved RBLN model
+# ----------------------------------------------------------------
+# Load the model we just saved - no need to recompile
+my_model_reloaded = RBLNResNetModel.from_pretrained("my_resnet_model_saved")
+print(my_model_reloaded)

{optimum_rbln-0.8.0.post1 → optimum_rbln-0.8.1}/pyproject.toml

@@ -28,12 +28,12 @@ classifiers = [
 ]
 keywords = ["transformers", "diffusers", "inference", "rbln", "atom", "rebel"]
 dependencies = [
-    "torch==2.6.0",
-    "torchaudio<=2.6.0",
-    "torchvision<=0.21.0",
+    "torch==2.7.0",
+    "torchaudio<=2.7.0",
+    "torchvision<=0.22.0",
     "accelerate>=1.0.1",
-    "transformers==4.50.3",
-    "diffusers<=0.31.0",
+    "transformers==4.51.3",
+    "diffusers==0.34.0",
     "packaging>=24.1",
 ]
 
@@ -78,6 +78,9 @@ version-file = "src/optimum/rbln/__version__.py"
 [tool.hatch.build.targets.wheel]
 packages = ["src/optimum"]
 
+[tool.hatch.metadata]
+allow-direct-references = true
+
 [tool.uv]
 environments = [
     "sys_platform == 'linux'"

{optimum_rbln-0.8.0.post1 → optimum_rbln-0.8.1}/src/optimum/rbln/__init__.py

@@ -26,6 +26,7 @@ _import_structure = {
         "RBLNModel",
     ],
     "configuration_utils": [
+        "RBLNAutoConfig",
         "RBLNCompileConfig",
         "RBLNModelConfig",
     ],
@@ -69,6 +70,8 @@ _import_structure = {
         "RBLNCLIPVisionModelConfig",
         "RBLNCLIPVisionModelWithProjection",
         "RBLNCLIPVisionModelWithProjectionConfig",
+        "RBLNColPaliForRetrieval",
+        "RBLNColPaliForRetrievalConfig",
         "RBLNDecoderOnlyModelForCausalLM",
         "RBLNDecoderOnlyModelForCausalLMConfig",
         "RBLNDistilBertForQuestionAnswering",
@@ -135,8 +138,17 @@ _import_structure = {
     "diffusers": [
         "RBLNAutoencoderKL",
         "RBLNAutoencoderKLConfig",
+        "RBLNAutoencoderKLCosmos",
+        "RBLNAutoencoderKLCosmosConfig",
         "RBLNControlNetModel",
         "RBLNControlNetModelConfig",
+        "RBLNCosmosTextToWorldPipeline",
+        "RBLNCosmosVideoToWorldPipeline",
+        "RBLNCosmosTextToWorldPipelineConfig",
+        "RBLNCosmosVideoToWorldPipelineConfig",
+        "RBLNCosmosSafetyChecker",
+        "RBLNCosmosTransformer3DModel",
+        "RBLNCosmosTransformer3DModelConfig",
         "RBLNDiffusionMixin",
         "RBLNKandinskyV22CombinedPipeline",
         "RBLNKandinskyV22CombinedPipelineConfig",
@@ -192,14 +204,24 @@ _import_structure = {
 
 if TYPE_CHECKING:
     from .configuration_utils import (
+        RBLNAutoConfig,
         RBLNCompileConfig,
         RBLNModelConfig,
     )
     from .diffusers import (
         RBLNAutoencoderKL,
         RBLNAutoencoderKLConfig,
+        RBLNAutoencoderKLCosmos,
+        RBLNAutoencoderKLCosmosConfig,
         RBLNControlNetModel,
         RBLNControlNetModelConfig,
+        RBLNCosmosSafetyChecker,
+        RBLNCosmosTextToWorldPipeline,
+        RBLNCosmosTextToWorldPipelineConfig,
+        RBLNCosmosTransformer3DModel,
+        RBLNCosmosTransformer3DModelConfig,
+        RBLNCosmosVideoToWorldPipeline,
+        RBLNCosmosVideoToWorldPipelineConfig,
         RBLNDiffusionMixin,
         RBLNKandinskyV22CombinedPipeline,
         RBLNKandinskyV22CombinedPipelineConfig,
@@ -295,6 +317,8 @@ if TYPE_CHECKING:
         RBLNCLIPVisionModelConfig,
         RBLNCLIPVisionModelWithProjection,
         RBLNCLIPVisionModelWithProjectionConfig,
+        RBLNColPaliForRetrieval,
+        RBLNColPaliForRetrievalConfig,
         RBLNDecoderOnlyModelForCausalLM,
         RBLNDecoderOnlyModelForCausalLMConfig,
         RBLNDistilBertForQuestionAnswering,

{optimum_rbln-0.8.0.post1 → optimum_rbln-0.8.1}/src/optimum/rbln/__version__.py

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