fix publishing

mhamilton723 · Feb 27, 2024 · 1d750f8 · 1d750f8
1 parent 2320bf5
commit 1d750f8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/build_and_upload.yml b/.github/workflows/build_and_upload.yml
@@ -8,6 +8,9 @@ on:
 jobs:
   build-and-upload:
     runs-on: ubuntu-latest
+    container:
+      image: nvidia/cuda:11.0-base  # Use an appropriate CUDA image
+      options: --gpus all
     steps:
     - uses: actions/checkout@v2
     - name: Set up Python
@@ -26,4 +29,3 @@ jobs:
       with:
         user: __token__
         password: ${{ secrets.PYPI_API_TOKEN }}
-
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name='featup',
-    version='0.1.0',
+    version='0.1.1',
     packages=find_packages(),
     install_requires=[
         'torch',