optimum-rbln 0.7.2rc1__tar.gz → 0.7.3__tar.gz

{optimum_rbln-0.7.2rc1 → optimum_rbln-0.7.3}/.github/scripts/auto_code_review.py

@@ -97,50 +97,6 @@ def remove_file_from_diff(diff_content, file_to_remove):
     return "\n".join(result)
 
 
-def skip_bot(pr):
-    global force_review
-    """
-    Review if
-    1. last commit messages starts with "[autoreview]"
-    2. last comment contains "/autoreview"
-    """
-
-    # Check commit message
-    commits = list(pr.get_commits())
-    if len(commits) == 0:
-        return True
-
-    last_commit = commits[-1]
-    try:
-        commit_message = last_commit.raw_data["commit"]["message"]
-    except KeyError:
-        commit_message = ""
-
-    if commit_message.startswith("[autoreview]"):
-        return False
-
-    # Check the last comment
-    comments = list(pr.get_issue_comments())
-    if len(comments) == 0:
-        return True
-
-    last = comments[-1]
-    if last.user.login.find("github-actions") != -1:
-        return True
-
-    if last.body.find("/autoreview") == -1:
-        return True
-
-    if last.reactions["heart"] > 0:
-        return True
-
-    if last.body.find("force") != -1:
-        force_review = True
-
-    last.create_reaction("heart")
-    return False
-
-
 def main():
     github_token = os.getenv("GITHUB_TOKEN")
     pr_number = os.getenv("PR_NUMBER")
@@ -155,13 +111,6 @@ def main():
     repo = g.get_repo(os.getenv("GITHUB_REPOSITORY"))
     pr = repo.get_pull(int(pr_number))
 
-    if skip_bot(pr):
-        print(
-            "To invoke review, Write '/autoreview' and re-run github actions,"
-            " or start the commit message with '[autoreview]'. "
-        )
-        sys.exit(0)
-
     # Get PR diff
     diff = get_pr_diff()
     diff = remove_file_from_diff(diff, "uv.lock")

optimum_rbln-0.7.3/.github/version.yaml

@@ -0,0 +1 @@
+rebel_compiler_version: 0.7.3.dev166+g05e9b26d

optimum_rbln-0.7.3/.github/workflows/auto_code_review.yml

@@ -0,0 +1,72 @@
+name: Auto Code Review
+
+on:
+  pull_request:
+  issue_comment:
+    types: [created]
+  push:
+    branches:
+      - '**'
+
+env:
+  GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
+  GOOGLE_MODEL_ID: ${{ vars.GOOGLE_MODEL_ID }}
+
+jobs:
+  auto-review:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v3
+        with:
+          fetch-depth: 2
+      
+      - name: Check if review should run
+        id: check
+        run: |
+          PR_NUMBER=""
+          SHOULD_RUN="false"
+          
+          # For push events, check commit message
+          if [[ "${{ github.event_name }}" == "push" ]]; then
+            if [[ "${{ contains(github.event.head_commit.message, '[autoreview]') }}" == "true" ]]; then
+              SHOULD_RUN="true"
+              # Use GitHub CLI to find PR associated with this commit
+              PR_NUMBER=$(gh pr list --head ${{ github.ref_name }} --json number --jq '.[0].number')
+            fi
+          
+          # For PR events
+          elif [[ "${{ github.event_name }}" == "pull_request" ]]; then
+            PR_NUMBER="${{ github.event.pull_request.number }}"
+          
+          # For comment events, check if it's "/autoreview"
+          elif [[ "${{ github.event_name }}" == "issue_comment" ]]; then
+            if [[ "${{ github.event.issue.pull_request != null }}" == "true" && "${{ contains(github.event.comment.body, '/autoreview') }}" == "true" ]]; then
+              SHOULD_RUN="true"
+              PR_NUMBER="${{ github.event.issue.number }}"
+            fi
+          fi
+          
+          echo "should_run=$SHOULD_RUN" >> $GITHUB_OUTPUT
+          echo "pr_number=$PR_NUMBER" >> $GITHUB_OUTPUT
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Set up Python
+        if: steps.check.outputs.should_run == 'true' && steps.check.outputs.pr_number != ''
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.x'
+
+      - name: Install dependencies
+        if: steps.check.outputs.should_run == 'true' && steps.check.outputs.pr_number != ''
+        run: |
+          python -m pip install --upgrade pip
+          pip install google-generativeai PyGithub
+
+      - name: Run Auto Code Review
+        if: steps.check.outputs.should_run == 'true' && steps.check.outputs.pr_number != ''
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          PR_NUMBER: ${{ steps.check.outputs.pr_number }}
+        run: python .github/scripts/auto_code_review.py

optimum_rbln-0.7.3/.github/workflows/deploy-on-tag.yaml

@@ -0,0 +1,34 @@
+name: Deploy package on tag released
+
+on:
+  push:
+    tags:
+      - 'v*'
+  workflow_dispatch:
+
+jobs:
+  deploy-on-tag:
+    uses: ./.github/workflows/deploy.yaml
+    with:
+      version: ${{ github.ref_name }}
+    secrets: inherit
+
+  trigger-vllm-model-compilation:
+    name: trigger vllm model compile and generation ci
+    needs: [deploy-on-tag]
+    runs-on: rebel-k8s-runner
+    steps:
+      - uses: actions/github-script@v6
+        with:
+          github-token: ${{ secrets.GIT_PAT }}
+          script: |
+            const result = await github.rest.actions.createWorkflowDispatch({
+              owner: 'rebellions-sw',
+              repo: 'rebel_compiler',
+              workflow_id: 'rebel_dispatch_model_generation_for_vllm.yaml',
+              ref: 'dev',
+              inputs: {
+                optimum_rbln_version: ${{ github.ref_name }},
+              }
+            })
+            console.log(result)

optimum_rbln-0.7.3/.github/workflows/rbln_check_compiler.yaml

@@ -0,0 +1,61 @@
+name: Check Rebel Compiler Version
+
+on:
+  workflow_call:
+    inputs:
+      compiler_version:
+        description: "Rebel compiler version to check"
+        required: true
+        type: string
+    outputs:
+      is_deploy_version:
+        description: "Whether the compiler is a deploy version"
+        value: ${{ jobs.check-rebel-compiler-version.outputs.is_deploy_version }}
+      compiler_version:
+        description: "The compiler version used"
+        value: ${{ jobs.check-rebel-compiler-version.outputs.compiler_version }}
+
+jobs:
+  check-rebel-compiler-version:
+    runs-on: rebel-k8s-runner
+    outputs:
+      is_deploy_version: ${{ steps.check_prod.outputs.IS_PROD }}
+      compiler_version: ${{ steps.install_compiler.outputs.COMPILER_VERSION }}
+    steps:
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.9"
+
+      - name: Install rebel-compiler
+        id: install_compiler
+        env:
+          REBEL_PYPI_ENDPOINT: ${{ vars.REBEL_PYPI_INTERNAL_ENDPOINT }}
+          REBEL_PYPI_USERNAME: ${{ secrets.REBEL_PYPI_USERNAME }}
+          REBEL_PYPI_PASSWORD: ${{ secrets.REBEL_PYPI_PASSWORD }}
+        run: |
+          set -e
+          PYPI_URL=$(echo $REBEL_PYPI_ENDPOINT | sed "s/\/\//\0$REBEL_PYPI_USERNAME:$REBEL_PYPI_PASSWORD@/")
+          pip3 install --extra-index-url $PYPI_URL rebel-compiler==${{ inputs.compiler_version }}
+          echo "COMPILER_VERSION=${{ inputs.compiler_version }}" >> $GITHUB_OUTPUT
+
+      - name: Run script to check ENV
+        id: check_env
+        run: |
+          echo "Running check for rebel-compiler version"
+          ENV_VALUE=$(python3 -c '${{ secrets.CHECK_DEPLOY }}')
+          echo $ENV_VALUE
+          echo "ENV_VALUE=$ENV_VALUE" >> $GITHUB_ENV
+
+      - name: Verify ENV is PROD
+        id: check_prod
+        run: |
+          if [ "$ENV_VALUE" = "PROD" ]; then
+            echo "IS_PROD=true" >> $GITHUB_OUTPUT
+            echo "version check pass(✅)!!"
+          else
+            echo "IS_PROD=false" >> $GITHUB_OUTPUT
+            echo "version check fail(❌)!!" 
+            echo "rebel-compiler must be prod version"
+            exit 1
+          fi 

{optimum_rbln-0.7.2rc1 → optimum_rbln-0.7.3}/.github/workflows/rbln_optimum_pytest.yaml

@@ -15,6 +15,11 @@ on:
         description: "rebel_compiler version to run"
         required: true
         type: string
+      test_level:
+        description: "Test level for OPTIMUM_RBLN_TEST_LEVEL (default, full, essential)"
+        required: false
+        type: string
+        default: "default"
 
 env:
   REBEL_PYPI_ENDPOINT: ${{ vars.REBEL_PYPI_INTERNAL_ENDPOINT }}
@@ -33,6 +38,13 @@ jobs:
           submodules: recursive
           fetch-depth: 0
 
+      - name: Get commit message if not provided
+        id: get_commit_message
+        if: ${{ inputs.commit_message == '' }}
+        run: |
+          COMMIT_MESSAGE=$(git log -1 --pretty=%B)
+          echo "message=$COMMIT_MESSAGE" >> $GITHUB_OUTPUT
+
       - name: Setup uv
         uses: astral-sh/setup-uv@v3
         with:
@@ -50,21 +62,34 @@ jobs:
         run: |
           PYPI_URL=$(echo ${{ env.REBEL_PYPI_ENDPOINT }} | sed "s/\/\//\0${{ env.REBEL_PYPI_USERNAME}}:${{ env.REBEL_PYPI_PASSWORD}}@/")
           uv pip install --extra-index-url $PYPI_URL rebel-compiler==${{ inputs.rebel_compiler_version }}
-
+      
       - name: Run pytest (transformers)
         env:
-          OPTIMUM_RBLN_TEST_LEVEL: default
+          OPTIMUM_RBLN_TEST_LEVEL: ${{ inputs.test_level }}
         run: |
-          uv run --no-sync pytest tests/test_transformers.py
+          echo 
+          if ${{ !contains( steps.get_commit_message.outputs.message , '[skip-transformers]') }}; then
+            uv run --no-sync pytest tests/test_transformers.py
+          else
+            echo "Found [skip-transformers] in commit message, skipping CI"
+          fi
 
       - name: Run pytest (diffusers)
         env:
-          OPTIMUM_RBLN_TEST_LEVEL: default
+          OPTIMUM_RBLN_TEST_LEVEL: ${{ inputs.test_level }}
         run: |
-          uv run --no-sync pytest tests/test_diffusers.py
+          if ${{ !contains( steps.get_commit_message.outputs.message , '[skip-diffusers]') }}; then
+            uv run --no-sync pytest tests/test_diffusers.py
+          else
+            echo "Found [skip-diffusers] in commit message, skipping CI"
+          fi
 
       - name: Run pytest (llm)
         env:
-          OPTIMUM_RBLN_TEST_LEVEL: default
+          OPTIMUM_RBLN_TEST_LEVEL: ${{ inputs.test_level }}
         run: |
-          uv run --no-sync pytest tests/test_llm.py      
+          if ${{ !contains( steps.get_commit_message.outputs.message , '[skip-llms]') }}; then
+            uv run --no-sync pytest tests/test_llm.py
+          else
+            echo "Found [skip-llms] in commit message, skipping CI"
+          fi

optimum_rbln-0.7.3/.github/workflows/rbln_scheduled_test.yaml

@@ -0,0 +1,51 @@
+name: Optimum-rbln / Scheduled Test
+
+on:
+  schedule:
+    # Run every day at 2am (17:00 UTC, 2:00am KST)
+    - cron: '0 17 * * *'
+
+env:
+  HF_USER_ID: ${{ secrets.HF_USER_ID }}
+  HF_AUTH_TOKEN: ${{ secrets.HF_AUTH_TOKEN }}
+
+jobs:
+  load-version:
+    runs-on: rebel-k8s-runner
+    outputs:
+      compiler_version: ${{ steps.get_version.outputs.compiler_version }}
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+      
+      - name: Get compiler version
+        id: get_version
+        run: |
+          VERSION=$(grep rebel_compiler_version .github/version.yaml | cut -d ':' -f2 | tr -d ' ')
+          echo "compiler_version=$VERSION" >> $GITHUB_OUTPUT
+
+  check-compiler:
+    needs: load-version
+    uses: ./.github/workflows/rbln_check_compiler.yaml
+    with:
+      compiler_version: ${{ needs.load-version.outputs.compiler_version }}
+    secrets: inherit
+
+  optimum-rbln-pytest:
+    needs: [load-version, check-compiler]
+    if: ${{ needs.check-compiler.outputs.is_deploy_version == 'true' }}
+    uses: ./.github/workflows/rbln_optimum_pytest.yaml
+    with:
+      ref: main
+      rebel_compiler_version: ${{ needs.check-compiler.outputs.compiler_version }}
+      test_level: "full"
+    secrets: inherit
+
+  optimum-rbln-inference-test:
+    needs: check-compiler
+    if: ${{ needs.check-compiler.outputs.is_deploy_version == 'true' }}
+    uses: ./.github/workflows/rbln_optimum_inference_test.yaml
+    with:
+      ref: main
+      rebel_compiler_version: ${{ needs.check-compiler.outputs.compiler_version }}
+    secrets: inherit

optimum_rbln-0.7.3/.github/workflows/rbln_trigger_on_pr.yaml

@@ -0,0 +1,94 @@
+name: Optimum-rbln / PR
+
+on:
+  pull_request:
+    branches:
+      - main
+
+env:
+  REBEL_PYPI_ENDPOINT: ${{ vars.REBEL_PYPI_INTERNAL_ENDPOINT }}
+  REBEL_PYPI_USERNAME: ${{ secrets.REBEL_PYPI_USERNAME }}
+  REBEL_PYPI_PASSWORD: ${{ secrets.REBEL_PYPI_PASSWORD }}
+
+jobs:
+  check-skip-ci:
+    runs-on: rebel-k8s-runner
+    outputs:
+      should_skip: ${{ contains(github.event.pull_request.head.commit.message, '[skip ci]') }}
+    steps:
+      - name: Check if [skip ci] is in commit message
+        run: |
+          if ${{ contains(github.event.pull_request.head.commit.message, '[skip ci]') }}; then
+            echo "Found [skip ci] in commit message, skipping CI"
+          else
+            echo "No [skip ci] found, continuing with CI"
+          fi
+
+  load-version:
+    runs-on: rebel-k8s-runner
+    needs: check-skip-ci
+    if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' }}
+    outputs:
+      compiler_version: ${{ steps.get_version.outputs.compiler_version }}
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+      
+      - name: Get compiler version
+        id: get_version
+        run: |
+          VERSION=$(grep rebel_compiler_version .github/version.yaml | cut -d ':' -f2 | tr -d ' ')
+          echo "compiler_version=$VERSION" >> $GITHUB_OUTPUT
+
+  check-compiler:
+    needs: [check-skip-ci, load-version]
+    if: ${{ needs.check-skip-ci.outputs.should_skip != 'true' }}
+    uses: ./.github/workflows/rbln_check_compiler.yaml
+    with:
+      compiler_version: ${{ needs.load-version.outputs.compiler_version }}
+    secrets: inherit
+
+  check-team-member:
+    runs-on: rebel-k8s-runner
+    needs: [check-skip-ci, 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 }}
+    steps:
+      - name: Fetch team members
+        id: fetch_team
+        run: |
+          response=$(curl -s -H "Authorization: Bearer ${{ secrets.GIT_PAT }}" \
+            -H "Content-Type: application/json" \
+            -d '{"query":"query { organization(login: \"rebellions-sw\") { team(slug: \"rebel-sw-team\") { members(first: 100) { nodes { login } } } } }"}' \
+            https://api.github.com/graphql)
+          echo "$response" | jq -r '.data.organization.team.members.nodes[].login' > team_members.txt
+
+      - name: Check if PR author is a team member
+        id: check_member
+        run: |
+          pr_author=${{ github.event.pull_request.user.login }}
+          if grep -qx "$pr_author" team_members.txt; then
+            echo "IS_TEAM_MEMBER=true" >> $GITHUB_OUTPUT
+          else
+            echo "IS_TEAM_MEMBER=false" >> $GITHUB_OUTPUT
+          fi
+
+  optimum-rbln-pytest:
+    needs: [check-skip-ci, 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:
+      ref: ${{ github.event.pull_request.head.sha }}
+      rebel_compiler_version: ${{ needs.check-compiler.outputs.compiler_version }}
+      test_level: "default"
+    secrets: inherit
+
+  optimum-rbln-inference-test:
+    needs: [check-skip-ci, 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:
+      ref: ${{ github.event.pull_request.head.sha }}
+      rebel_compiler_version: ${{ needs.check-compiler.outputs.compiler_version }}
+    secrets: inherit

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimum-rbln
-Version: 0.7.2rc1
+Version: 0.7.3
 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.3/examples/kandinsky2_2/run_kandinsky2_2.py

@@ -0,0 +1,57 @@
+import os
+
+import fire
+import torch
+
+from optimum.rbln import RBLNKandinskyV22Pipeline, RBLNKandinskyV22PriorPipeline
+
+
+def main(
+    prior_model_id: str = "kandinsky-community/kandinsky-2-2-prior",
+    inpaint_model_id: str = "kandinsky-community/kandinsky-2-2-decoder",
+    from_diffusers: bool = False,
+    prompt: str = "red cat, 4k photo",
+):
+    if from_diffusers:
+        prior_pipe = RBLNKandinskyV22PriorPipeline.from_pretrained(
+            model_id=prior_model_id,
+            export=True,
+        )
+        prior_pipe.save_pretrained(os.path.basename(prior_model_id))
+
+        pipe = RBLNKandinskyV22Pipeline.from_pretrained(
+            model_id=inpaint_model_id,
+            export=True,
+            rbln_img_height=768,
+            rbln_img_width=768,
+        )
+        pipe.save_pretrained(os.path.basename(inpaint_model_id))
+    else:
+        prior_pipe = RBLNKandinskyV22PriorPipeline.from_pretrained(
+            model_id=os.path.basename(prior_model_id),
+            export=False,
+        )
+        pipe = RBLNKandinskyV22Pipeline.from_pretrained(
+            model_id=os.path.basename(inpaint_model_id),
+            export=False,
+        )
+
+    generator = torch.manual_seed(42)
+    out = prior_pipe(prompt, generator=generator)
+    image_emb = out.image_embeds
+    zero_image_emb = out.negative_image_embeds
+
+    out = pipe(
+        image_embeds=image_emb,
+        negative_image_embeds=zero_image_emb,
+        height=768,
+        width=768,
+        num_inference_steps=50,
+        generator=generator,
+    )
+    image = out.images[0]
+    image.save(f"{prompt}.png")
+
+
+if __name__ == "__main__":
+    fire.Fire(main)

optimum_rbln-0.7.3/examples/kandinsky2_2/run_kandinsky2_2_combined.py

@@ -0,0 +1,31 @@
+import os
+
+import fire
+import torch
+
+from optimum.rbln import RBLNKandinskyV22CombinedPipeline
+
+
+def main(
+    model_id: str = "kandinsky-community/kandinsky-2-2-decoder",
+    from_diffusers: bool = False,
+    prompt: str = "red cat, 4k photo",
+):
+    if from_diffusers:
+        pipe = RBLNKandinskyV22CombinedPipeline.from_pretrained(
+            model_id=model_id,
+            export=True,
+            rbln_img_height=768,
+            rbln_img_width=768,
+        )
+        pipe.save_pretrained(os.path.basename(model_id))
+    else:
+        pipe = RBLNKandinskyV22CombinedPipeline.from_pretrained(model_id=os.path.basename(model_id), export=False)
+
+    generator = torch.manual_seed(42)
+    image = pipe(prompt, height=768, width=768, num_inference_steps=50, generator=generator).images[0]
+    image.save(f"{prompt}.png")
+
+
+if __name__ == "__main__":
+    fire.Fire(main)

optimum_rbln-0.7.3/examples/kandinsky2_2/run_kandinsky2_2_img2img.py

@@ -0,0 +1,61 @@
+import os
+
+import fire
+import torch
+from diffusers.utils import load_image
+
+from optimum.rbln import RBLNKandinskyV22Img2ImgPipeline, RBLNKandinskyV22PriorPipeline
+
+
+def main(
+    prior_model_id: str = "kandinsky-community/kandinsky-2-2-prior",
+    inpaint_model_id: str = "kandinsky-community/kandinsky-2-2-decoder",
+    from_diffusers: bool = False,
+    prompt: str = "A red cartoon frog, 4k",
+):
+    img_url = "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/kandinsky/frog.png"
+    init_image = load_image(img_url)
+
+    if from_diffusers:
+        prior_pipe = RBLNKandinskyV22PriorPipeline.from_pretrained(
+            model_id=prior_model_id,
+            export=True,
+        )
+        prior_pipe.save_pretrained(os.path.basename(prior_model_id))
+
+        pipe = RBLNKandinskyV22Img2ImgPipeline.from_pretrained(
+            model_id=inpaint_model_id,
+            export=True,
+            rbln_img_height=768,
+            rbln_img_width=768,
+        )
+        pipe.save_pretrained(os.path.basename(inpaint_model_id))
+    else:
+        prior_pipe = RBLNKandinskyV22PriorPipeline.from_pretrained(
+            model_id=os.path.basename(prior_model_id),
+            export=False,
+        )
+        pipe = RBLNKandinskyV22Img2ImgPipeline.from_pretrained(
+            model_id=os.path.basename(inpaint_model_id),
+            export=False,
+        )
+
+    generator = torch.manual_seed(42)
+    image_emb, zero_image_emb = prior_pipe(prompt, generator=generator, return_dict=False)
+
+    out = pipe(
+        image=init_image,
+        image_embeds=image_emb,
+        negative_image_embeds=zero_image_emb,
+        height=768,
+        width=768,
+        num_inference_steps=100,
+        strength=0.2,
+        generator=generator,
+    )
+    image = out.images[0]
+    image.save(f"{prompt}.png")
+
+
+if __name__ == "__main__":
+    fire.Fire(main)

optimum_rbln-0.7.3/examples/kandinsky2_2/run_kandinsky2_2_img2img_combined.py

@@ -0,0 +1,46 @@
+import os
+
+import fire
+import torch
+from diffusers.utils import load_image
+
+from optimum.rbln import RBLNKandinskyV22Img2ImgCombinedPipeline
+
+
+def main(
+    model_id: str = "kandinsky-community/kandinsky-2-2-decoder",
+    from_diffusers: bool = False,
+    prompt: str = "A red cartoon frog, 4k",
+):
+    img_url = "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/kandinsky/frog.png"
+    init_image = load_image(img_url)
+
+    if from_diffusers:
+        pipe = RBLNKandinskyV22Img2ImgCombinedPipeline.from_pretrained(
+            model_id=model_id,
+            export=True,
+            rbln_img_height=768,
+            rbln_img_width=768,
+        )
+        pipe.save_pretrained(os.path.basename(model_id))
+    else:
+        pipe = RBLNKandinskyV22Img2ImgCombinedPipeline.from_pretrained(
+            model_id=os.path.basename(model_id), export=False
+        )
+
+    generator = torch.manual_seed(42)
+
+    image = pipe(
+        prompt=prompt,
+        image=init_image,
+        height=768,
+        width=768,
+        num_inference_steps=100,
+        strength=0.2,
+        generator=generator,
+    ).images[0]
+    image.save(f"{prompt}.png")
+
+
+if __name__ == "__main__":
+    fire.Fire(main)