mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
c89bdfbe72
* Put models in subfolders * Styling * Fix imports in tests * More fixes in test imports * Sneaky hidden imports * Fix imports in doc files * More sneaky imports * Finish fixing tests * Fix examples * Fix path for copies * More fixes for examples * Fix dummy files * More fixes for example * More model import fixes * Is this why you're unhappy GitHub? * Fix imports in conver command
45 lines
1.2 KiB
YAML
45 lines
1.2 KiB
YAML
name: Torch hub integration
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- "*"
|
|
|
|
jobs:
|
|
torch_hub_integration:
|
|
runs-on: ubuntu-latest
|
|
env:
|
|
# TODO quickfix but may need more investigation
|
|
ACTIONS_ALLOW_UNSECURE_COMMANDS: True
|
|
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 tokenizers packaging
|
|
|
|
- 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'))"
|