Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
81 changes: 20 additions & 61 deletions src/common/models/models.ts
Original file line number Diff line number Diff line change
Expand Up @@ -919,54 +919,56 @@ const MODEL_REGISTRY: ModelConfig[] = [

/**
* Image generation model registry
* Only includes models with TEXT input and IMAGE output modalities
*/
const IMAGE_GENERATION_MODELS: ModelConfig[] = [
// Stability AI models
// Amazon models
{
baseId: 'stability.sd3-5-large-v1:0',
name: 'Stability SD3.5 Large',
provider: 'stability',
baseId: 'amazon.nova-canvas-v1:0',
name: 'Amazon Nova Canvas',
provider: 'amazon',
category: 'image',
toolUse: false,
maxTokensLimit: 0,
inferenceProfiles: [
{
type: 'base',
regions: ['us-west-2']
regions: ['us-east-1', 'us-west-2', 'ap-northeast-1', 'eu-west-1']
}
]
},
{
baseId: 'stability.sd3-large-v1:0',
name: 'Stability SD3 Large',
provider: 'stability',
baseId: 'amazon.titan-image-generator-v2:0',
name: 'Amazon Titan Image Generator v2',
provider: 'amazon',
category: 'image',
toolUse: false,
maxTokensLimit: 0,
inferenceProfiles: [
{
type: 'base',
regions: ['us-west-2']
regions: ['us-east-1', 'us-west-2']
}
]
},
{
baseId: 'stability.stable-image-core-v1:0',
name: 'Stability Stable Image Core v1.0',
provider: 'stability',
baseId: 'amazon.titan-image-generator-v1',
name: 'Amazon Titan Image Generator v1',
provider: 'amazon',
category: 'image',
toolUse: false,
maxTokensLimit: 0,
inferenceProfiles: [
{
type: 'base',
regions: ['us-west-2']
regions: ['us-east-1', 'us-west-2']
}
]
},
// Stability AI models
{
baseId: 'stability.stable-image-core-v1:1',
name: 'Stability Stable Image Core v1.1',
baseId: 'stability.sd3-5-large-v1:0',
name: 'Stable Diffusion 3.5 Large',
provider: 'stability',
category: 'image',
toolUse: false,
Expand All @@ -979,8 +981,8 @@ const IMAGE_GENERATION_MODELS: ModelConfig[] = [
]
},
{
baseId: 'stability.stable-image-ultra-v1:0',
name: 'Stability Stable Image Ultra v1.0',
baseId: 'stability.stable-image-core-v1:1',
name: 'Stable Image Core',
provider: 'stability',
category: 'image',
toolUse: false,
Expand All @@ -994,7 +996,7 @@ const IMAGE_GENERATION_MODELS: ModelConfig[] = [
},
{
baseId: 'stability.stable-image-ultra-v1:1',
name: 'Stability Stable Image Ultra v1.1',
name: 'Stable Image Ultra',
provider: 'stability',
category: 'image',
toolUse: false,
Expand All @@ -1005,49 +1007,6 @@ const IMAGE_GENERATION_MODELS: ModelConfig[] = [
regions: ['us-west-2']
}
]
},
// Amazon models
{
baseId: 'amazon.nova-canvas-v1:0',
name: 'Amazon Nova Canvas',
provider: 'amazon',
category: 'image',
toolUse: false,
maxTokensLimit: 0,
inferenceProfiles: [
{
type: 'base',
regions: ['us-east-1', 'ap-northeast-1', 'eu-west-1']
}
]
},
{
baseId: 'amazon.titan-image-generator-v2:0',
name: 'Amazon Titan Image Generator v2',
provider: 'amazon',
category: 'image',
toolUse: false,
maxTokensLimit: 0,
inferenceProfiles: [
{
type: 'base',
regions: ['us-east-1', 'us-west-2']
}
]
},
{
baseId: 'amazon.titan-image-generator-v1',
name: 'Amazon Titan Image Generator v1',
provider: 'amazon',
category: 'image',
toolUse: false,
maxTokensLimit: 0,
inferenceProfiles: [
{
type: 'base',
regions: ['us-east-1', 'us-west-2', 'eu-west-1', 'eu-west-2', 'ap-south-1']
}
]
}
]

Expand Down
4 changes: 3 additions & 1 deletion src/renderer/src/i18n/locales/en.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ import { chatPage } from './chat'
import { awsDiagramGenerator } from './awsDiagramGenerator'
import { stepFunctionGenerator } from './stepFunctionGenerator'
import { websiteGenerator } from './websiteGenerator'
import { imageGenerator } from './imageGenerator'
import {
iamPolicy,
notificationSettings,
Expand Down Expand Up @@ -981,7 +982,8 @@ const en = {
backgroundAgent: BackgroundAgent,
ignoreSettings: IgnoreSettings,
projectIgnore: ProjectIgnore,
common: Common
common: Common,
imageGenerator: imageGenerator.en
}

export default en
44 changes: 44 additions & 0 deletions src/renderer/src/i18n/locales/imageGenerator/index.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
export const imageGenerator = {
en: {
title: 'Image Generator',
prompt: 'Prompt',
promptPlaceholder: 'Describe the image you want to generate...',
promptRequired: 'Please enter a prompt',
negativePrompt: 'Negative Prompt',
negativePromptPlaceholder: 'Things to exclude from the image...',
aspectRatio: 'Aspect Ratio',
outputFormat: 'Output Format',
modelInfo: 'Model',
generate: 'Generate Image',
generating: 'Generating...',
success: 'Image generated successfully',
error: 'Failed to generate image',
noImage: 'Generated images will appear here',
hint: 'Press ⌘+Enter to generate',
history: 'Recent',
settingsTitle: 'Image Generation',
settingsDescription: 'Configure the default model for image generation',
defaultModel: 'Default Model'
},
ja: {
title: '画像生成',
prompt: 'プロンプト',
promptPlaceholder: '生成したい画像を説明してください...',
promptRequired: 'プロンプトを入力してください',
negativePrompt: 'ネガティブプロンプト',
negativePromptPlaceholder: '画像から除外したいもの...',
aspectRatio: 'アスペクト比',
outputFormat: '出力形式',
modelInfo: 'モデル',
generate: '画像を生成',
generating: '生成中...',
success: '画像が正常に生成されました',
error: '画像の生成に失敗しました',
noImage: '生成された画像がここに表示されます',
hint: '⌘+Enterで生成',
history: '履歴',
settingsTitle: '画像生成',
settingsDescription: '画像生成のデフォルトモデルを設定',
defaultModel: 'デフォルトモデル'
}
}
4 changes: 3 additions & 1 deletion src/renderer/src/i18n/locales/ja.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ import { chatPage } from './chat'
import { awsDiagramGenerator } from './awsDiagramGenerator'
import { stepFunctionGenerator } from './stepFunctionGenerator'
import { websiteGenerator } from './websiteGenerator'
import { imageGenerator } from './imageGenerator'
import {
iamPolicy,
notificationSettings,
Expand Down Expand Up @@ -1001,7 +1002,8 @@ const ja = {
backgroundAgent: BackgroundAgent,
ignoreSettings: IgnoreSettings,
projectIgnore: ProjectIgnore,
common: Common
common: Common,
imageGenerator: imageGenerator.ja
}

export default ja
Loading
Loading