From ffed150070aeea0e6203d98e2988aefe30de1c6f Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 2 Jul 2026 06:37:32 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/inference/diffusion.py | 1 + studio/backend/core/inference/diffusion_families.py | 4 +--- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/studio/backend/core/inference/diffusion.py b/studio/backend/core/inference/diffusion.py index 978432f8be..eb4bb33034 100644 --- a/studio/backend/core/inference/diffusion.py +++ b/studio/backend/core/inference/diffusion.py @@ -1042,6 +1042,7 @@ class DiffusionBackend: install_arch_patches, uninstall_arch_patches, ) + try: if effective_speed != SPEED_OFF: install_compile_safe_patches() diff --git a/studio/backend/core/inference/diffusion_families.py b/studio/backend/core/inference/diffusion_families.py index b95278b037..08140f7957 100644 --- a/studio/backend/core/inference/diffusion_families.py +++ b/studio/backend/core/inference/diffusion_families.py @@ -258,9 +258,7 @@ def _token_in_needle(token: str, needle: str) -> bool: ('qwen-image-edit' in 'qwen-image-edit-2511') while preventing a short token from matching inside an unrelated word ('kontext' must not match 'kontextual', 'edit' must not match 'edition').""" - return re.search( - r"(?:^|[-_./\\])" + re.escape(token) + r"(?:$|[-_./\\])", needle - ) is not None + return re.search(r"(?:^|[-_./\\])" + re.escape(token) + r"(?:$|[-_./\\])", needle) is not None def _best_family_match(needle: str) -> Optional[DiffusionFamily]: