From e5dddfe0b65de0937d179ff0f200515c3e35ab8e Mon Sep 17 00:00:00 2001 From: Artem Chumachenko Date: Thu, 11 Apr 2024 11:37:27 +0200 Subject: [PATCH] Try mps device selecting --- .github/workflows/run-tests.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run-tests.yaml b/.github/workflows/run-tests.yaml index d6316d48b..32baf1141 100644 --- a/.github/workflows/run-tests.yaml +++ b/.github/workflows/run-tests.yaml @@ -14,8 +14,8 @@ jobs: - { model: 'bigscience/bloom-560m', os: 'ubuntu', python-version: '3.11' } - { model: 'Maykeye/TinyLLama-v0', os: 'ubuntu', python-version: '3.8' } - { model: 'Maykeye/TinyLLama-v0', os: 'ubuntu', python-version: '3.11' } - - { model: 'Maykeye/TinyLLama-v0', os: 'macos', python-version: '3.10' } - - { model: 'Maykeye/TinyLLama-v0', os: 'macos', python-version: '3.11' } + - { model: 'Maykeye/TinyLLama-v0', os: 'macos-14', python-version: '3.10' } + - { model: 'Maykeye/TinyLLama-v0', os: 'macos-14', python-version: '3.11' } - { model: 'artek0chumak/TestMixtral', os: 'ubuntu', python-version: '3.8' } - { model: 'artek0chumak/TestMixtral', os: 'ubuntu', python-version: '3.11' } fail-fast: false