From c87f846f07713e8ddc43791c2d3100222afcf3e0 Mon Sep 17 00:00:00 2001
From: Alexander Grattan <alex@grattan.me>
Date: Fri, 26 Jul 2024 12:18:24 -0400
Subject: [PATCH] feat: use bytedance sdxl lightning model

---
 apps/server/src/services/generation.service.ts | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/apps/server/src/services/generation.service.ts b/apps/server/src/services/generation.service.ts
index a9ed921..46e56df 100644
--- a/apps/server/src/services/generation.service.ts
+++ b/apps/server/src/services/generation.service.ts
@@ -238,9 +238,15 @@ function isReplicateResponse(res: object): res is [string, string] {
 
 export async function getReplicateAIImages({ prompt }: { prompt: string }) {
   const output = await replicate.run(
-    "stability-ai/sdxl:7762fd07cf82c948538e41f63f77d685e02b063e37e496e96eefd46c929f9bdc",
+    "bytedance/sdxl-lightning-4step:5f24084160c9089501c1b3545d9be3c27883ae2239b6f412990e82d4a6210f8f",
     {
-      input: { prompt, num_outputs: 2, width: 768, height: 768 },
+      input: {
+        prompt,
+        negative_prompt: "worst quality, low quality",
+        num_outputs: 2,
+        width: 768,
+        height: 768,
+      },
     }
   );