diff --git a/src/api/controllers/images.ts b/src/api/controllers/images.ts index 41d6106..5a0f251 100644 --- a/src/api/controllers/images.ts +++ b/src/api/controllers/images.ts @@ -11,6 +11,9 @@ const DEFAULT_ASSISTANT_ID = "513695"; export const DEFAULT_MODEL = "jimeng-3.0"; const DRAFT_VERSION = "3.0.2"; const MODEL_MAP = { + "jimeng-4.5": "high_aes_general_v40l", + "jimeng-4.1": "high_aes_general_v41", + "jimeng-4.0": "high_aes_general_v40", "jimeng-3.1": "high_aes_general_v30l_art_fangzhou:general_v3.0_18b", "jimeng-3.0": "high_aes_general_v30l:general_v3.0_18b", "jimeng-2.1": "high_aes_general_v21_L:general_v2.1_L", @@ -46,6 +49,9 @@ export async function generateImages( try { const model = getModel(_model); + const isV4 = _model.startsWith('jimeng-4'); + const currentVersion = isV4 ? '3.3.7' : DRAFT_VERSION; + const resolutionType = isV4 ? '2k' : '1k'; logger.info(`使用模型: ${_model} 映射模型: ${model} ${width}x${height} 精细度: ${sampleStrength}`); const { totalCredit } = await getCredit(refreshToken); @@ -87,7 +93,7 @@ export async function generateImages( id: util.uuid(), min_version: DRAFT_VERSION, is_from_tsn: true, - version: DRAFT_VERSION, + version: currentVersion, main_component_id: componentId, component_list: [ { @@ -116,7 +122,7 @@ export async function generateImages( id: util.uuid(), height, width, - resolution_type: "1k", + resolution_type: resolutionType, }, }, history_option: { @@ -312,6 +318,8 @@ export async function uploadImages( refreshToken: string ) { const model = getModel(_model); + const isV4 = _model.startsWith('jimeng-4'); + const currentVersion = isV4 ? '3.3.7' : DRAFT_VERSION; logger.info(`使用模型: ${_model} 映射模型: ${model} ${width}x${height} 精细度: ${sampleStrength}`); const { totalCredit } = await getCredit(refreshToken); @@ -353,7 +361,7 @@ export async function uploadImages( id: util.uuid(), min_version: DRAFT_VERSION, is_from_tsn: true, - version: DRAFT_VERSION, + version: currentVersion, main_component_id: componentId, component_list: [ {