[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
d0c5cf6e07
commit
c34020bfec
2 changed files with 4 additions and 1 deletions
|
|
@ -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),
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue