Update pre-commit hooks (#12475)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-01-28 00:23:08 +00:00
committed by GitHub
parent 6116ca8cd7
commit 823ab79633
64 changed files with 322 additions and 288 deletions

View File

@@ -1,6 +1,6 @@
import os
from contextlib import contextmanager
from functools import lru_cache
from functools import cache
from typing import Generator, Optional, Type
import torch
@@ -100,7 +100,7 @@ def get_attn_backend(
)
@lru_cache(maxsize=None)
@cache
def _cached_get_attn_backend(
head_size: int,
dtype: torch.dtype,