optimum-rbln 0.8.1a5__tar.gz → 0.8.1a7__tar.gz

optimum_rbln-0.8.1a7/.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.1a5 → optimum_rbln-0.8.1a7}/.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.1a5 → optimum_rbln-0.8.1a7}/.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.1a5 → optimum_rbln-0.8.1a7}/.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.1a7/.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.1a5 → optimum_rbln-0.8.1a7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimum-rbln
-Version: 0.8.1a5
+Version: 0.8.1a7
 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/examples/cosmos/run_cosmos_text2world.py

@@ -0,0 +1,70 @@
+import os
+
+import fire
+from diffusers.utils import export_to_video
+
+from optimum.rbln import RBLNCosmosTextToWorldPipeline
+
+
+def main(
+    model_id: str = "nvidia/Cosmos-1.0-Diffusion-7B-Text2World",
+    from_diffusers: bool = False,
+    prompt: str = None,
+    steps: int = 36,
+    height: int = 704,
+    width: int = 1280,
+):
+    if prompt is None:
+        prompt = "A sleek, humanoid robot stands in a vast warehouse filled with neatly stacked cardboard boxes on industrial shelves. The robot's metallic body gleams under the bright, even lighting, highlighting its futuristic design and intricate joints. A glowing blue light emanates from its chest, adding a touch of advanced technology. The background is dominated by rows of boxes, suggesting a highly organized storage system. The floor is lined with wooden pallets, enhancing the industrial setting. The camera remains static, capturing the robot's poised stance amidst the orderly environment, with a shallow depth of field that keeps the focus on the robot while subtly blurring the background for a cinematic effect."
+    print(prompt)
+
+    if from_diffusers:
+        pipe = RBLNCosmosTextToWorldPipeline.from_pretrained(
+            model_id,
+            export=True,
+            rbln_height=height,
+            rbln_width=width,
+            rbln_config={
+                "transformer": {
+                    "device": [4, 5, 6, 7],
+                    "tensor_parallel_size": 4,
+                },
+                "text_encoder": {
+                    "device": 1,
+                },
+                "vae": {
+                    "device": 2,
+                },
+                "safety_checker": {
+                    "aegis": {"device": [0, 1, 2, 3]},
+                },
+            },
+        )
+        pipe.save_pretrained(os.path.basename(model_id))
+    else:
+        pipe = RBLNCosmosTextToWorldPipeline.from_pretrained(
+            os.path.basename(model_id),
+            export=False,
+            rbln_config={
+                "transformer": {
+                    "device": [4, 5, 6, 7],
+                    "tensor_parallel_size": 4,
+                },
+                "text_encoder": {
+                    "device": 1,
+                },
+                "vae": {
+                    "device": 2,
+                },
+                "safety_checker": {
+                    "aegis": {"device": [0, 1, 2, 3]},
+                },
+            },
+        )
+
+    output = pipe(prompt=prompt, num_inference_steps=steps).frames[0]
+    export_to_video(output, "output.mp4", fps=30)
+
+
+if __name__ == "__main__":
+    fire.Fire(main)

{optimum_rbln-0.8.1a5 → optimum_rbln-0.8.1a7}/pyproject.toml

@@ -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.1a5 → optimum_rbln-0.8.1a7}/src/optimum/rbln/__init__.py

@@ -138,8 +138,17 @@ _import_structure = {
     "diffusers": [
         "RBLNAutoencoderKL",
         "RBLNAutoencoderKLConfig",
+        "RBLNAutoencoderKLCosmos",
+        "RBLNAutoencoderKLCosmosConfig",
         "RBLNControlNetModel",
         "RBLNControlNetModelConfig",
+        "RBLNCosmosTextToWorldPipeline",
+        "RBLNCosmosVideoToWorldPipeline",
+        "RBLNCosmosTextToWorldPipelineConfig",
+        "RBLNCosmosVideoToWorldPipelineConfig",
+        "RBLNCosmosSafetyChecker",
+        "RBLNCosmosTransformer3DModel",
+        "RBLNCosmosTransformer3DModelConfig",
         "RBLNDiffusionMixin",
         "RBLNKandinskyV22CombinedPipeline",
         "RBLNKandinskyV22CombinedPipelineConfig",
@@ -202,8 +211,17 @@ if TYPE_CHECKING:
     from .diffusers import (
         RBLNAutoencoderKL,
         RBLNAutoencoderKLConfig,
+        RBLNAutoencoderKLCosmos,
+        RBLNAutoencoderKLCosmosConfig,
         RBLNControlNetModel,
         RBLNControlNetModelConfig,
+        RBLNCosmosSafetyChecker,
+        RBLNCosmosTextToWorldPipeline,
+        RBLNCosmosTextToWorldPipelineConfig,
+        RBLNCosmosTransformer3DModel,
+        RBLNCosmosTransformer3DModelConfig,
+        RBLNCosmosVideoToWorldPipeline,
+        RBLNCosmosVideoToWorldPipelineConfig,
         RBLNDiffusionMixin,
         RBLNKandinskyV22CombinedPipeline,
         RBLNKandinskyV22CombinedPipelineConfig,

{optimum_rbln-0.8.1a5 → optimum_rbln-0.8.1a7}/src/optimum/rbln/__version__.py

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

{optimum_rbln-0.8.1a5 → optimum_rbln-0.8.1a7}/src/optimum/rbln/diffusers/__init__.py

@@ -18,14 +18,21 @@ from diffusers.pipelines.pipeline_utils import ALL_IMPORTABLE_CLASSES, LOADABLE_
 from transformers.utils import _LazyModule
 
 
-LOADABLE_CLASSES["optimum.rbln"] = {"RBLNBaseModel": ["save_pretrained", "from_pretrained"]}
+LOADABLE_CLASSES["optimum.rbln"] = {
+    "RBLNBaseModel": ["save_pretrained", "from_pretrained"],
+    "RBLNCosmosSafetyChecker": ["save_pretrained", "from_pretrained"],
+}
 ALL_IMPORTABLE_CLASSES.update(LOADABLE_CLASSES["optimum.rbln"])
 
 
 _import_structure = {
     "configurations": [
         "RBLNAutoencoderKLConfig",
+        "RBLNAutoencoderKLCosmosConfig",
         "RBLNControlNetModelConfig",
+        "RBLNCosmosTextToWorldPipelineConfig",
+        "RBLNCosmosVideoToWorldPipelineConfig",
+        "RBLNCosmosTransformer3DModelConfig",
         "RBLNKandinskyV22CombinedPipelineConfig",
         "RBLNKandinskyV22Img2ImgCombinedPipelineConfig",
         "RBLNKandinskyV22Img2ImgPipelineConfig",
@@ -52,6 +59,9 @@ _import_structure = {
         "RBLNVQModelConfig",
     ],
     "pipelines": [
+        "RBLNCosmosTextToWorldPipeline",
+        "RBLNCosmosVideoToWorldPipeline",
+        "RBLNCosmosSafetyChecker",
         "RBLNKandinskyV22CombinedPipeline",
         "RBLNKandinskyV22Img2ImgCombinedPipeline",
         "RBLNKandinskyV22InpaintCombinedPipeline",
@@ -76,8 +86,10 @@ _import_structure = {
     ],
     "models": [
         "RBLNAutoencoderKL",
+        "RBLNAutoencoderKLCosmos",
         "RBLNUNet2DConditionModel",
         "RBLNControlNetModel",
+        "RBLNCosmosTransformer3DModel",
         "RBLNSD3Transformer2DModel",
         "RBLNPriorTransformer",
         "RBLNVQModel",
@@ -90,7 +102,11 @@ _import_structure = {
 if TYPE_CHECKING:
     from .configurations import (
         RBLNAutoencoderKLConfig,
+        RBLNAutoencoderKLCosmosConfig,
         RBLNControlNetModelConfig,
+        RBLNCosmosTextToWorldPipelineConfig,
+        RBLNCosmosTransformer3DModelConfig,
+        RBLNCosmosVideoToWorldPipelineConfig,
         RBLNKandinskyV22CombinedPipelineConfig,
         RBLNKandinskyV22Img2ImgCombinedPipelineConfig,
         RBLNKandinskyV22Img2ImgPipelineConfig,
@@ -120,12 +136,16 @@ if TYPE_CHECKING:
     from .models import (
         RBLNAutoencoderKL,
         RBLNControlNetModel,
+        RBLNCosmosTransformer3DModel,
         RBLNPriorTransformer,
         RBLNSD3Transformer2DModel,
         RBLNUNet2DConditionModel,
         RBLNVQModel,
     )
     from .pipelines import (
+        RBLNCosmosSafetyChecker,
+        RBLNCosmosTextToWorldPipeline,
+        RBLNCosmosVideoToWorldPipeline,
         RBLNKandinskyV22CombinedPipeline,
         RBLNKandinskyV22Img2ImgCombinedPipeline,
         RBLNKandinskyV22Img2ImgPipeline,

{optimum_rbln-0.8.1a5 → optimum_rbln-0.8.1a7}/src/optimum/rbln/diffusers/configurations/__init__.py

@@ -1,12 +1,16 @@
 from .models import (
     RBLNAutoencoderKLConfig,
+    RBLNAutoencoderKLCosmosConfig,
     RBLNControlNetModelConfig,
+    RBLNCosmosTransformer3DModelConfig,
     RBLNPriorTransformerConfig,
     RBLNSD3Transformer2DModelConfig,
     RBLNUNet2DConditionModelConfig,
     RBLNVQModelConfig,
 )
 from .pipelines import (
+    RBLNCosmosTextToWorldPipelineConfig,
+    RBLNCosmosVideoToWorldPipelineConfig,
     RBLNKandinskyV22CombinedPipelineConfig,
     RBLNKandinskyV22Img2ImgCombinedPipelineConfig,
     RBLNKandinskyV22Img2ImgPipelineConfig,

{optimum_rbln-0.8.1a5 → optimum_rbln-0.8.1a7}/src/optimum/rbln/diffusers/configurations/models/__init__.py

@@ -1,5 +1,7 @@
 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_sd3 import RBLNSD3Transformer2DModelConfig
 from .configuration_unet_2d_condition import RBLNUNet2DConditionModelConfig