fix style & typo
This commit is contained in:
parent
e62ac4d63a
commit
5fb8c275c3
|
@ -160,7 +160,7 @@ jobs:
|
|||
|
||||
if [[ ${{ inputs.hardware }} == "rocm" ]]
|
||||
then
|
||||
echo "docker_volume=/data/cache/.cache/huggingface/hub"
|
||||
echo "docker_volume=/data/cache/.cache/huggingface/hub" >> "$GITHUB_OUTPUT"
|
||||
else
|
||||
echo "docker_volume=/mnt/cache" >> "$GITHUB_OUTPUT"
|
||||
fi
|
||||
|
|
|
@ -4,6 +4,7 @@ from testing_utils import require_backend_async, require_backend
|
|||
|
||||
# These tests do not pass on ROCm, that does not support head_dim > 128 (2b model is 256).
|
||||
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
@require_backend("cuda", "xpu")
|
||||
def flash_gemma_gptq_handle(launcher):
|
||||
|
|
|
@ -51,6 +51,7 @@ def is_flaky_async(
|
|||
|
||||
return decorator
|
||||
|
||||
|
||||
def require_backend(*args):
|
||||
def decorator(func):
|
||||
@functools.wraps(func)
|
||||
|
|
Loading…
Reference in New Issue