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]: