Try mps device selecting

bigscience-workshop · Apr 11, 2024 · e5dddfe · e5dddfe
1 parent f02bd57
commit e5dddfe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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