fix style & typo

This commit is contained in:
Felix Marty 2024-06-20 09:03:00 +00:00 committed by Nicolas Patry
parent e62ac4d63a
commit 5fb8c275c3
No known key found for this signature in database
GPG Key ID: E939E8CC91A1C674
3 changed files with 3 additions and 1 deletions

View File

@ -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

View File

@ -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):

View File

@ -51,6 +51,7 @@ def is_flaky_async(
return decorator
def require_backend(*args):
def decorator(func):
@functools.wraps(func)