add flux-1 to choices and use it as default

spongedsc · Nov 14, 2024 · 82a9b77 · 82a9b77
1 parent 0d075c4
commit 82a9b77
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/commands/imagine.js b/src/commands/imagine.js
@@ -11,6 +11,10 @@ export default {
 				.setName("model")
 				.setDescription("Enter a description of what you're generating.")
 				.setChoices([
+					{
+						name: "black-forest-labs/flux-1-schnell",
+						value: "@cf/black-forest-labs/flux-1-schnell",
+					},
 					{
 						name: "lykon/dreamshaper-8-lcm",
 						value: "@cf/lykon/dreamshaper-8-lcm",

diff --git a/src/util/models/interactions.js b/src/util/models/interactions.js
@@ -137,7 +137,7 @@ export class InteractionResponse {
 		const callToModel = await this.workersAI
 			.callModel(
 				{
-					model: "@cf/lykon/dreamshaper-8-lcm",
+					model: "@cf/black-forest-labs/flux-1-schnell",
 					input: {
 						prompt: data,
 					},
@@ -376,7 +376,7 @@ export class InteractionMessageEvent {
 					content: this.response.formatAssistantMessage(`\n${genData.trim()}`, "imagine"),
 					contextId: this.message?.id,
 					respondingTo: this.message?.id,
-					model: "@cf/lykon/dreamshaper-8-lcm",
+					model: "@cf/black-forest-labs/flux-1-schnell",
 				},
 				true,
 			)