Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Jul 23, 2024
1 parent f8a8bba commit 4d9220f
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions .github/workflows/test_cli_mps_pytorch.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,6 @@ jobs:
- name: Run tests
env:
PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
FORCE_SERIAL: 1
run: |
python -c "from transformers import AutoModel; model = AutoModel.from_pretrained('bert-base-uncased').to('mps'); print(model)"
pytest -s -k "cli and mps and pytorch"

0 comments on commit 4d9220f

Please sign in to comment.