[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-07-01 20:02:34 +00:00
commit c34020bfec
2 changed files with 4 additions and 1 deletions

View file

@ -182,7 +182,9 @@ def detect_family(repo_id: str, override: Optional[str] = None) -> Optional[Diff
def detect_family_for_pick(
repo_id: str, gguf_filename: Optional[str] = None, override: Optional[str] = None
repo_id: str,
gguf_filename: Optional[str] = None,
override: Optional[str] = None,
) -> Optional[DiffusionFamily]:
"""``detect_family``, falling back to the combined path/filename for a direct
local ``.gguf`` pick. The frontend splits such a pick into (parent dir, basename),

View file

@ -253,6 +253,7 @@ def _enable_cudnn_benchmark(logger: Any) -> bool:
def _enable_tf32(logger: Any) -> bool:
try:
import torch
torch.backends.cuda.matmul.allow_tf32 = True
torch.backends.cudnn.allow_tf32 = True
return True