From a4bc511460101a4720405c28c5d539d9bf85625f Mon Sep 17 00:00:00 2001
From: dontAskVI <villem.ossip@gmail.com>
Date: Wed, 11 Dec 2024 17:05:44 +0200
Subject: [PATCH] Add image model for Galadriel model provider

---
 docs/api/enumerations/ModelProviderName.md | 4 ++--
 packages/core/src/models.ts                | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/docs/api/enumerations/ModelProviderName.md b/docs/api/enumerations/ModelProviderName.md
index c0983bed02..6c1f6c1363 100644
--- a/docs/api/enumerations/ModelProviderName.md
+++ b/docs/api/enumerations/ModelProviderName.md
@@ -146,9 +146,9 @@ Available model providers
 
 ---
 
-### GALADRIEL
+### 
 
-> **GALADRIEL**: `"galadriel"`
+> ****: `""`
 
 #### Defined in
 
diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts
index 7f397d280d..f386b217cc 100644
--- a/packages/core/src/models.ts
+++ b/packages/core/src/models.ts
@@ -328,6 +328,7 @@ export const models: Models = {
             [ModelClass.MEDIUM]: "llama3.1:70b",
             [ModelClass.LARGE]: "llama3.1:405b",
             [ModelClass.EMBEDDING]: "gte-large-en-v1.5",
+            [ModelClass.IMAGE]: "stabilityai/stable-diffusion-xl-base-1.0",
         },
     },
     [ModelProviderName.FAL]: {