|
|
|
@ -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: 'tiiuae/falcon-rw-1b', os: 'ubuntu', python-version: '3.8' }
|
|
|
|
|
- { model: 'tiiuae/falcon-rw-1b', os: 'ubuntu', python-version: '3.11' }
|
|
|
|
|
- { model: 'petals-team/falcon-rw-1b', os: 'ubuntu', python-version: '3.8' }
|
|
|
|
|
- { model: 'petals-team/falcon-rw-1b', 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' }
|
|
|
|
|
fail-fast: false
|
|
|
|
@ -95,6 +95,9 @@ jobs:
|
|
|
|
|
|
|
|
|
|
# [Step 2] Run PyTest
|
|
|
|
|
|
|
|
|
|
# Share disk cache between Petals servers, clients, and HF Transformers
|
|
|
|
|
export TRANSFORMERS_CACHE=~/.cache/petals
|
|
|
|
|
|
|
|
|
|
# Necessary for @pytest.mark.forked to work properly on macOS, see https://github.com/kevlened/pytest-parallel/issues/93
|
|
|
|
|
export no_proxy=*
|
|
|
|
|
export OBJC_DISABLE_INITIALIZE_FORK_SAFETY=YES
|
|
|
|
|