Workaround for removed Docker image

See replicate/cog#1162.
garfieldnate · Jul 4, 2023 · 3d842e2 · 3d842e2
1 parent 5527273
commit 3d842e2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cog.yaml b/cog.yaml
@@ -1,5 +1,7 @@
 build:
   gpu: true
+  # see https://github.com/replicate/cog/issues/1162
+  cuda: "11.1.1"
   python_version: "3.11"
   python_packages:
     - "stable-ts==2.0.0"