display different cfg and step values depending on model

huggingface · Aug 29, 2024 · e23ea90 · e23ea90
1 parent 7886513
commit e23ea90
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 [