mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
42 lines
1.1 KiB
YAML
42 lines
1.1 KiB
YAML
name: Torch hub integration
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- "*"
|
|
|
|
jobs:
|
|
torch_hub_integration:
|
|
runs-on: ubuntu-latest
|
|
steps:
|
|
# no checkout necessary here.
|
|
- name: Extract branch name
|
|
run: echo "::set-env name=BRANCH::${GITHUB_REF#refs/heads/}"
|
|
- name: Check branch name
|
|
run: echo $BRANCH
|
|
- name: Set up Python
|
|
uses: actions/setup-python@v1
|
|
with:
|
|
python-version: 3.7
|
|
|
|
- name: Loading cache
|
|
uses: actions/cache@v2
|
|
id: cache
|
|
with:
|
|
path: ~/.cache/pip
|
|
key: v0-torch_hub-${{ hashFiles('setup.py') }}
|
|
|
|
- name: Install dependencies
|
|
run: |
|
|
pip install --upgrade pip
|
|
pip install torch
|
|
pip install numpy filelock protobuf requests tqdm regex sentencepiece sacremoses packaging
|
|
pip install tokenizers==0.9.0.rc2
|
|
|
|
- name: Torch hub list
|
|
run: |
|
|
python -c "import torch; print(torch.hub.list('huggingface/transformers:$BRANCH'))"
|
|
|
|
- name: Torch hub help
|
|
run: |
|
|
python -c "import torch; print(torch.hub.help('huggingface/transformers:$BRANCH', 'modelForSequenceClassification'))"
|