Fix segment anything default model ID (#2317)

livepeer · Sep 27, 2024 · 6715c2a · 6715c2a
1 parent 9ab67ff
commit 6715c2a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/packages/api/src/schema/ai-api-schema.yaml b/packages/api/src/schema/ai-api-schema.yaml
@@ -544,7 +544,7 @@ components:
           type: string
           title: Model Id
           description: Hugging Face model ID used for image generation.
-          default: 'facebook/sam2-hiera-large:'
+          default: 'facebook/sam2-hiera-large'
         point_coords:
           type: string
           title: Point Coords

diff --git a/packages/api/src/schema/pull-ai-schema.js b/packages/api/src/schema/pull-ai-schema.js
@@ -11,7 +11,7 @@ export const defaultModels = {
   "image-to-video": "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
   upscale: "stabilityai/stable-diffusion-x4-upscaler",
   "audio-to-text": "openai/whisper-large-v3",
-  "segment-anything-2": "facebook/sam2-hiera-large:",
+  "segment-anything-2": "facebook/sam2-hiera-large",
 };
 const schemaDir = path.resolve(__dirname, ".");
 const aiSchemaUrl =