Update packages/tasks/src/model-libraries-snippets.ts

apolinario · mishig25 · web-flow · commit 15983c516a12 · 2024-09-19T05:11:10.000-05:00
Co-authored-by: Mishig &lt;dmishig@gmail.com&gt;
diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts
@@ -74,7 +74,10 @@ function get_base_diffusers_model(model: ModelData): string {
 }
 
 function get_prompt_from_diffusers_model(model: ModelData): string | undefined {
-	return (model.widgetData?.[0] as WidgetExampleTextInput).text ?? model.cardData?.instance_prompt;
+	const prompt = (model.widgetData?.[0] as WidgetExampleTextInput).text ?? model.cardData?.instance_prompt;
+    if(prompt){
+        return escapeQuotes(prompt);
+    }
 }
 
 export const bertopic = (model: ModelData): string[] => [

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,10 @@ function get_base_diffusers_model(model: ModelData): string {`
`74`	`74`	`}`
`75`	`75`
`76`	`76`	`function get_prompt_from_diffusers_model(model: ModelData): string \| undefined {`
`77`		`- return (model.widgetData?.[0] as WidgetExampleTextInput).text ?? model.cardData?.instance_prompt;`
	`77`	`+ const prompt = (model.widgetData?.[0] as WidgetExampleTextInput).text ?? model.cardData?.instance_prompt;`
	`78`	`+ if(prompt){`
	`79`	`+ return escapeQuotes(prompt);`
	`80`	`+ }`
`78`	`81`	`}`
`79`	`82`
`80`	`83`	`export const bertopic = (model: ModelData): string[] => [`