|
@@ -18,7 +18,7 @@ jobs:
|
|
|
id: cache-model
|
|
|
uses: actions/cache@v3
|
|
|
with:
|
|
|
- path: ~/.dummy
|
|
|
+ path: ~/converted_ok
|
|
|
key: model-v1-${{ hashFiles('setup.cfg', 'src/petals/cli/convert_model.py') }}
|
|
|
- name: Set up Python
|
|
|
if: steps.cache-model.outputs.cache-hit != 'true'
|
|
@@ -52,7 +52,7 @@ jobs:
|
|
|
export HF_TAG=${{ hashFiles('setup.cfg', 'src/petals/cli/convert_model.py') }}
|
|
|
python -m petals.cli.convert_model --model bigscience/bloom-560m --output_path ./converted_model \
|
|
|
--output_repo bloom-testing/test-bloomd-560m-$HF_TAG --use_auth_token $BLOOM_TESTING_WRITE_TOKEN \
|
|
|
- --resize_token_embeddings 50000
|
|
|
+ --resize_token_embeddings 50000 && touch ~/converted_ok
|
|
|
|
|
|
run-tests:
|
|
|
runs-on: ubuntu-latest
|