display different cfg and step values depending on model

huggingface · Aug 30, 2024 · 0436d98 · 0436d98
1 parent b40fa66
commit 0436d98
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
@@ -137,7 +137,8 @@ const snippetDiffusionKit = (model: ModelData): LocalAppSnippet[] => {
 			`  --height 768 \\`,
 			`  --width 1360 \\`,
 			`  --seed 1001 \\`,
-			`  --step 4 \\`,
+			`  --step ${model.tags.includes("flux") ? 4 : 50} \\`,
+			`  --cfg ${model.tags.includes("flux") ? 0. : 5} \\`,
 			`  --output ~/Desktop/out.png`,
 		].join("\n");
 	return [