From 7e940581995d00156931b995c1b9efed69103282 Mon Sep 17 00:00:00 2001 From: kangfenmao Date: Tue, 4 Feb 2025 12:28:17 +0800 Subject: [PATCH] feat: update text-to-image model regex to include Janus --- src/renderer/src/config/models.ts | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index 8abc5dcdab..69d74aae89 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -144,8 +144,7 @@ const visionAllowedModels = [ 'pixtral', 'gpt-4(?:-[\\w-]+)', 'gpt-4o(?:-[\\w-]+)?', - 'chatgpt-4o(?:-[\\w-]+)?', - 'janus' + 'chatgpt-4o(?:-[\\w-]+)?' ] const visionExcludedModels = ['gpt-4-\\d+-preview', 'gpt-4-turbo-preview', 'gpt-4-32k', 'gpt-4-\\d+'] @@ -155,7 +154,7 @@ export const VISION_REGEX = new RegExp( 'i' ) -export const TEXT_TO_IMAGE_REGEX = /flux|diffusion|stabilityai|sd-|dall|cogview/i +export const TEXT_TO_IMAGE_REGEX = /flux|diffusion|stabilityai|sd-|dall|cogview|janus/i export const EMBEDDING_REGEX = /(?:^text-|embed|rerank|davinci|babbage|bge-|e5-|LLM2Vec|retrieval|uae-|gte-|jina)/i export const NOT_SUPPORTED_REGEX = /(?:^tts|rerank|whisper|speech)/i