Update tasks specs (automated commit)

huggingface · Feb 5, 2025 · acf9eb1 · acf9eb1
1 parent 493ee5f
commit acf9eb1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/packages/tasks/src/tasks/feature-extraction/inference.ts b/packages/tasks/src/tasks/feature-extraction/inference.ts
@@ -13,9 +13,9 @@ export type FeatureExtractionOutput = Array<number[]>;
  */
 export interface FeatureExtractionInput {
 	/**
-	 * The text or list of texts to embed.
+	 * The text to embed.
 	 */
-	inputs: FeatureExtractionInputs;
+	inputs: string;
 	normalize?: boolean;
 	/**
 	 * The name of the prompt that should be used by for encoding. If not set, no prompt
@@ -34,8 +34,4 @@ export interface FeatureExtractionInput {
 	truncation_direction?: FeatureExtractionInputTruncationDirection;
 	[property: string]: unknown;
 }
-/**
- * The text or list of texts to embed.
- */
-export type FeatureExtractionInputs = string[] | string;
 export type FeatureExtractionInputTruncationDirection = "Left" | "Right";
diff --git a/packages/tasks/src/tasks/feature-extraction/spec/input.json b/packages/tasks/src/tasks/feature-extraction/spec/input.json
@@ -7,12 +7,8 @@
 	"required": ["inputs"],
 	"properties": {
 		"inputs": {
-			"title": "FeatureExtractionInputs",
-			"oneOf": [
-				{ "type": "string" },
-				{ "type": "array", "items": { "type": "string" } }
-			],
-			"description": "The text or list of texts to embed."
+			"type": "string",
+			"description": "The text to embed."
 		},
 		"normalize": {
 			"type": "boolean",