diff --git a/studio/backend/core/inference/diffusion.py b/studio/backend/core/inference/diffusion.py index ab453faab3..195ded40a6 100644 --- a/studio/backend/core/inference/diffusion.py +++ b/studio/backend/core/inference/diffusion.py @@ -375,12 +375,15 @@ def detect_family( # ``qwenimage-gguf`` -> qwenimage-compact in needle_compact. # Use word boundary on the compact form too: the compact # ``flux2`` must not match inside ``flux20``. - return bool( - re.search( - rf"(^|[^0-9a-z]){re.escape(term_compact)}([^0-9a-z]|$)", - needle_compact, + return ( + bool( + re.search( + rf"(^|[^0-9a-z]){re.escape(term_compact)}([^0-9a-z]|$)", + needle_compact, + ) ) - ) or term_compact == needle_compact + or term_compact == needle_compact + ) return False # Scan _FAMILIES first (GGUF-supported), then _FULL_REPO_FAMILIES diff --git a/studio/backend/models/inference.py b/studio/backend/models/inference.py index 8594b6306e..2dbd6bfc40 100644 --- a/studio/backend/models/inference.py +++ b/studio/backend/models/inference.py @@ -1455,9 +1455,7 @@ import re as _re _EMBEDDED_HF_TOKEN_RE = _re.compile(r"hf_[A-Za-z0-9]{20,}") -def _reject_embedded_hf_token( - value: Optional[str], field_name: str -) -> Optional[str]: +def _reject_embedded_hf_token(value: Optional[str], field_name: str) -> Optional[str]: """Refuse identifiers that contain an embedded ``hf_xxx`` token. Round 15 P1 #5: ``repo_id`` and ``base_repo`` accept URL-style