Fix get_prompt_from_diffusers_model

huggingface · Sep 20, 2024 · f975978 · f975978
1 parent c488b19
commit f975978
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts
@@ -74,7 +74,7 @@ function get_base_diffusers_model(model: ModelData): string {
 }
 
 function get_prompt_from_diffusers_model(model: ModelData): string | undefined {
-	const prompt = (model.widgetData?.[0] as WidgetExampleTextInput).text ?? model.cardData?.instance_prompt;
+	const prompt = (model.widgetData?.[0] as WidgetExampleTextInput | undefined)?.text ?? model.cardData?.instance_prompt;
 	if (prompt) {
 		return escapeStringForJson(prompt);
 	}