Skip to content

Use weights_only for load (#127) #443

Use weights_only for load (#127)

Use weights_only for load (#127) #443

Workflow file for this run

name: Test Installation
on:
push:
branches:
- main
pull_request:
branches:
- main
jobs:
test-installation:
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v2
- name: Set up Python 3.8
uses: actions/setup-python@v2
with:
python-version: 3.8
- name: Install dependencies
run: |
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
pip install -e .
- name: Test import
run: |
python -c "import segment_anything_fast; from segment_anything_fast import sam_model_registry"