Add stabilityai_pytorch release image definition (#3262)

aws · Aug 17, 2023 · ac26937 · ac26937
1 parent 8ae3e37
commit ac26937
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/.release_templates/stabilityai_release_images.yml b/.release_templates/stabilityai_release_images.yml
@@ -0,0 +1,16 @@
+---
+release_images:
+  1:
+    framework: "stabilityai_pytorch"
+    version: "2.0.1"
+    arch_type: "x86"
+    customer_type: "sagemaker"
+    inference:
+      device_types: ["gpu"]
+      python_versions: ["py310"]
+      os_version: "ubuntu20.04"
+      cuda_version: "cu118"
+      sgm_version: "0.1.0"
+      example: False        
+      disable_sm_tag: False
+      force_release: False
diff --git a/release_images_inference.yml b/release_images_inference.yml
@@ -24,3 +24,17 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
+  3:
+    framework: "stabilityai_pytorch"
+    version: "2.0.1"
+    arch_type: "x86"
+    customer_type: "sagemaker"
+    inference:
+      device_types: ["gpu"]
+      python_versions: ["py310"]
+      os_version: "ubuntu20.04"
+      cuda_version: "cu118"
+      sgm_version: "0.1.0"
+      example: False        
+      disable_sm_tag: False
+      force_release: False