add scheduler param

huggingface · Jan 26, 2024 · 9ccb3a4 · 9ccb3a4
1 parent 8dc4d17
commit 9ccb3a4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/packages/tasks/src/tasks/text-to-image/spec/input.json b/packages/tasks/src/tasks/text-to-image/spec/input.json
@@ -45,6 +45,10 @@
 						}
 					},
 					"required": ["width", "height"]
+				},
+				"scheduler": {
+					"type": "string",
+					"description": "For diffusion models. Override the scheduler with a compatible one"
 				}
 			}
 		}