From 82a9b77a28b23dcf7339c0e67ed950d949ad9c8f Mon Sep 17 00:00:00 2001
From: Arti <59352535+artifishvr@users.noreply.github.com>
Date: Wed, 13 Nov 2024 18:21:31 -0800
Subject: [PATCH] add flux-1 to choices and use it as default

---
 src/commands/imagine.js         | 4 ++++
 src/util/models/interactions.js | 4 ++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/commands/imagine.js b/src/commands/imagine.js
index 5bd485a..717ce5a 100644
--- 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
index ac24889..87e8d79 100644
--- 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,
 			)