From 80285686ae5f295b166cddd0be3c7bf2102fb1b6 Mon Sep 17 00:00:00 2001
From: Zeke Sikelianos <zeke@sikelianos.com>
Date: Thu, 16 May 2024 15:50:45 -0700
Subject: [PATCH 1/4] do not install CUDA by default

Update action.yml
---
 README.md  | 2 +-
 action.yml | 8 +++++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 15bf602..a76a588 100644
--- a/README.md
+++ b/README.md
@@ -63,7 +63,7 @@ Install NVIDIA CUDA Toolkit. Defaults to `true`. This flag can be set to `false`
 - name: Setup Cog
   uses: replicate/setup-cog@v1
   with:
-    install-cuda: false
+    install-cuda: true
 ```
 
 ## Example: Manual Deployment
diff --git a/action.yml b/action.yml
index 2ab341b..ce696cb 100644
--- a/action.yml
+++ b/action.yml
@@ -17,8 +17,10 @@ inputs:
   install-cuda:
     description: "Install NVIDIA CUDA Toolkit to build GPU-bound containers."
     required: false
-    default: 'true'
-
+    default: 'false'
+    options:
+      - true
+      - false
 
 runs:
   using: "composite"
@@ -27,7 +29,7 @@ runs:
         uses: docker/setup-buildx-action@d70bba72b1f3fd22344832f00baa16ece964efeb # v3.3.0
   
       - name: Install NVIDIA CUDA Toolkit
-        if: ${{ inputs.install-cuda == 'true' }}
+        if: inputs.install-cuda == true || inputs.install-cuda == 'true'
         uses: Jimver/cuda-toolkit@dc0ca7bb29c5a92f7a963d3d5c93f8d59765136a # v0.2.14
 
       - name: Setup Cog

From e872f219c5b59449c637bf651018b6105ce32843 Mon Sep 17 00:00:00 2001
From: Zeke Sikelianos <zeke@sikelianos.com>
Date: Thu, 16 May 2024 15:50:55 -0700
Subject: [PATCH 2/4] brackets not required

---
 action.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/action.yml b/action.yml
index ce696cb..85ad460 100644
--- a/action.yml
+++ b/action.yml
@@ -41,7 +41,7 @@ runs:
 
       - name: Log in to Replicate
         shell: bash
-        if: ${{ inputs.token != '' }}
+        if: inputs.token != ''
         env:
           REPLICATE_API_TOKEN: ${{ inputs.token }}
         run: |

From 584ed723339b14e89451e9df2fd96e20e470c9c0 Mon Sep 17 00:00:00 2001
From: Zeke Sikelianos <zeke@sikelianos.com>
Date: Thu, 16 May 2024 15:51:09 -0700
Subject: [PATCH 3/4] prepare docs for v2

---
 README.md | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/README.md b/README.md
index a76a588..5c2dbf4 100644
--- a/README.md
+++ b/README.md
@@ -12,7 +12,7 @@ Add a step to your workflow that uses this action:
 
 ```yml
 - name: Setup Cog
-  uses: replicate/setup-cog@v1
+  uses: replicate/setup-cog@v2
   with:
     token: ${{ secrets.REPLICATE_API_TOKEN }}
 
@@ -37,7 +37,7 @@ Defaults to `v0.8.6`.
 
 ```yml
 - name: Setup Cog
-  uses: replicate/setup-cog@v1
+  uses: replicate/setup-cog@v2
   with:
     cog-version: "v0.9.5"
 ```
@@ -48,7 +48,7 @@ Your Replicate API token. If set, the Action will automatically authenticate to
 
 ```yml
 - name: Setup Cog
-  uses: replicate/setup-cog@v1
+  uses: replicate/setup-cog@v2
   with:
     token: ${{ secrets.REPLICATE_API_TOKEN }}
 ```
@@ -61,7 +61,7 @@ Install NVIDIA CUDA Toolkit. Defaults to `true`. This flag can be set to `false`
 
 ```yml
 - name: Setup Cog
-  uses: replicate/setup-cog@v1
+  uses: replicate/setup-cog@v2
   with:
     install-cuda: true
 ```
@@ -88,7 +88,7 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Setup Cog
-        uses: replicate/setup-cog@v1
+        uses: replicate/setup-cog@v2
         with:
           token: ${{ secrets.REPLICATE_API_TOKEN }}
 
@@ -117,7 +117,7 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Setup Cog
-        uses: replicate/setup-cog@v1
+        uses: replicate/setup-cog@v2
         with:
           token: ${{ secrets.REPLICATE_API_TOKEN }}
 
@@ -149,5 +149,5 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Setup Cog
-        uses: replicate/setup-cog@v1
+        uses: replicate/setup-cog@v2
 ```

From 56d6b9eff0ececba2c6460110074b4a8587d4063 Mon Sep 17 00:00:00 2001
From: Zeke Sikelianos <zeke@sikelianos.com>
Date: Thu, 16 May 2024 15:54:34 -0700
Subject: [PATCH 4/4] update default cog version to 0.9.7

Update README.md
---
 README.md  | 4 ++--
 action.yml | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 5c2dbf4..35e41c5 100644
--- a/README.md
+++ b/README.md
@@ -33,13 +33,13 @@ This following inputs can be specified using the `with` keyword:
 
 The version of Cog to install in the form `vX.Y.Z`. See [Cog releases](https://github.com/replicate/cog/releases) for a list of available versions.
 
-Defaults to `v0.8.6`.
+Defaults to `v0.9.7`.
 
 ```yml
 - name: Setup Cog
   uses: replicate/setup-cog@v2
   with:
-    cog-version: "v0.9.5"
+    cog-version: "v0.9.8-alpha"
 ```
 
 #### `token`
diff --git a/action.yml b/action.yml
index 85ad460..bc71dcd 100644
--- a/action.yml
+++ b/action.yml
@@ -13,7 +13,7 @@ inputs:
   cog-version:
     description: "The version of Cog to install"
     required: false
-    default: "v0.8.6"
+    default: "v0.9.7"
   install-cuda:
     description: "Install NVIDIA CUDA Toolkit to build GPU-bound containers."
     required: false