ci: add Ruff config, pre-commit hooks, and GitHub Actions lint workflow

.github/workflows/lint.yml

@@ -0,0 +1,17 @@
+name: Lint
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: astral-sh/setup-uv@v5
+      - run: uv python install 3.10
+      - run: uv pip install ruff
+      - run: uv run ruff check .
+      - run: uv run ruff format --check .

app.py

@@ -15,14 +15,12 @@
 import gc
 
 import gradio as gr
-import pillow_avif
 import torch
 from huggingface_hub import snapshot_download
 from pillow_heif import register_heif_opener
 
 from pipelines.pipeline_infu_flux import InfUFluxPipeline
 
-
 # Register HEIF support for Pillow
 register_heif_opener()
 
@@ -80,9 +78,9 @@ def prepare_pipeline(model_version, enable_realism, enable_anti_blur):
         torch.cuda.empty_cache()
 
         if model_version == 'aes_stage2':
-            model_path = f'./models/InfiniteYou/infu_flux_v1.0/aes_stage2'
+            model_path = './models/InfiniteYou/infu_flux_v1.0/aes_stage2'
         elif model_version == 'sim_stage1':
-            model_path = f'./models/InfiniteYou/infu_flux_v1.0/sim_stage1'
+            model_path = './models/InfiniteYou/infu_flux_v1.0/sim_stage1'
         else:
             raise ValueError(f'Model version {model_version} not supported.')
         print(f'Loading model from {model_path}')

pipelines/pipeline_flux_infusenet.py

@@ -14,16 +14,15 @@
 # limitations under the License.
 
 import inspect
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Union
 
 import numpy as np
 import torch
 from diffusers import FluxControlNetPipeline
-from diffusers.models.controlnet_flux import FluxControlNetModel, FluxMultiControlNetModel
 from diffusers.image_processor import PipelineImageInput
+from diffusers.models.controlnet_flux import FluxControlNetModel, FluxMultiControlNetModel
 from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
-from diffusers.utils import replace_example_docstring, is_torch_xla_available, logging
-
+from diffusers.utils import is_torch_xla_available, logging
 
 if is_torch_xla_available():
     import torch_xla.core.xla_model as xm

pipelines/pipeline_infu_flux.py

@@ -25,8 +25,8 @@
 from huggingface_hub import snapshot_download
 from insightface.app import FaceAnalysis
 from insightface.utils import face_align
-from PIL import Image
 from optimum.quanto import freeze, qint8, quantize
+from PIL import Image
 from transformers import T5EncoderModel
 
 from .pipeline_flux_infusenet import FluxInfuseNetPipeline

pyproject.toml

@@ -0,0 +1,12 @@
+[tool.ruff]
+line-length = 120
+
+[tool.ruff.lint]
+select = ["E", "F", "I"]
+ignore = ["E501", "E402", "E731", "E722", "E741", "E721", "E701", "F841", "F403", "F405"]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]
+
+[tool.uv]
+# Install with: uv pip install -r requirements.txt