Update CircleCI configuration and mask generator

Former-commit-id: 5ac76991bdfb09511a4127b392ea06e38f758762

.circleci/config.yml

@@ -1,34 +1,17 @@
-# This config was automatically generated from your source code
-# Stacks detected: deps:python:.
 version: 2.1
+
 orbs:
   python: circleci/python@2
+
 jobs:
-  test-python:
-    # Install dependencies and run tests
+  install-dependencies:
     docker:
-      - image: cimg/python:3.8-node
+      - image: cimg/python:3.10-slim
     steps:
       - checkout
       - python/install-packages
-      - run:
-          name: Run tests
-          command: pytest --junitxml=junit.xml || ((($? == 5)) && echo 'Did not find any tests to run.')
-      - store_test_results:
-          path: junit.xml
-  deploy:
-    # This is an example deploy job, not actually used by the workflow
-    docker:
-      - image: cimg/base:stable
-    steps:
-      # Replace this with steps to deploy to users
-      - run:
-          name: deploy
-          command: '#e.g. ./deploy.sh'
+
 workflows:
-  build-and-test:
+  install-python-deps:
     jobs:
-      - test-python
-    # - deploy:
-    #     requires:
-    #       - test-python
+      - install-dependencies

logs/app_debug.log

@@ -1468,3 +1468,5 @@ speed: {'preprocess': 1.9655227661132812, 'inference': 86.20810508728027, 'postp
 2024-03-26 19:58:18,633 [INFO] models - I2VGenXL pipeline Inference ->
 2024-03-26 20:22:04,466 [INFO] clear_memory - Memory Cleared
 2024-03-26 20:22:04,471 [INFO] models - Stable Video Diffusion Image 2 Video pipeline Inference ->
+2024-03-26 20:53:11,563 [INFO] clear_memory - Memory Cleared
+2024-03-26 21:01:23,790 [INFO] clear_memory - Memory Cleared

logs/app_info.log

@@ -1468,3 +1468,5 @@ speed: {'preprocess': 1.9655227661132812, 'inference': 86.20810508728027, 'postp
 2024-03-26 19:58:18,633 [INFO] models - I2VGenXL pipeline Inference ->
 2024-03-26 20:22:04,466 [INFO] clear_memory - Memory Cleared
 2024-03-26 20:22:04,471 [INFO] models - Stable Video Diffusion Image 2 Video pipeline Inference ->
+2024-03-26 20:53:11,563 [INFO] clear_memory - Memory Cleared
+2024-03-26 21:01:23,790 [INFO] clear_memory - Memory Cleared

scripts/mask_generator.py

@@ -1,9 +1,7 @@
-from PIL import Image, ImageFilter,ImageDraw
+from PIL import Image
 import numpy as np
-from logger import rich_logger as l
-from ultralytics import YOLO
-import cv2
 import PIL.ImageOps
+from diffusers.utils import load_image
 
 
 

scripts/run.py

@@ -4,8 +4,6 @@ from mask_generator import  invert_mask
 from segment_everything import generate_mask_from_bbox
 from models import kandinsky_inpainting_inference, load_image
 from PIL import Image
-from pipeline import fetch_kandinsky_pipeline
-from config import controlnet_adapter_model_name, controlnet_base_model_name, kandinsky_model_name
 from mask_generator import extend_image
 
 
@@ -15,7 +13,7 @@ def main(args):
     os.makedirs(args.mask_dir, exist_ok=True)
 
     # Extend image
-    extended_image = extend_image(image_path=args.image_path, target_width=2560, target_height=1440, roi_scale=0.6)
+    extended_image = extend_image(image_path=args.image_path, target_width=2560, target_height=1440, roi_scale=0.7)
     extended_image_path = os.path.join(args.output_dir, f'{args.uid}_extended.jpg')
     extended_image.save(extended_image_path)