liuzheng/jimeng-free-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

4.x

Browse Source
This commit is contained in:
jonathang4 2025-12-12 19:57:51 +08:00
parent e9454d095c
commit 6654a0e076
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/api/controllers/images.ts
View File

@ -11,6 +11,9 @@ const DEFAULT_ASSISTANT_ID = "513695";
export const DEFAULT_MODEL = "jimeng-3.0"; export const DEFAULT_MODEL = "jimeng-3.0";
const DRAFT_VERSION = "3.0.2"; const DRAFT_VERSION = "3.0.2";
const MODEL_MAP = { 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.1": "high_aes_general_v30l_art_fangzhou:general_v3.0_18b",
"jimeng-3.0": "high_aes_general_v30l: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", "jimeng-2.1": "high_aes_general_v21_L:general_v2.1_L",
@ -46,6 +49,9 @@ export async function generateImages(
try { try {
const model = getModel(_model); 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}`); logger.info(`使用模型: ${_model} 映射模型: ${model} ${width}x${height} 精细度: ${sampleStrength}`);
const { totalCredit } = await getCredit(refreshToken); const { totalCredit } = await getCredit(refreshToken);
@ -87,7 +93,7 @@ export async function generateImages(
id: util.uuid(), id: util.uuid(),
min_version: DRAFT_VERSION, min_version: DRAFT_VERSION,
is_from_tsn: true, is_from_tsn: true,
version: DRAFT_VERSION, version: currentVersion,
main_component_id: componentId, main_component_id: componentId,
component_list: [ component_list: [
{ {
@ -116,7 +122,7 @@ export async function generateImages(
id: util.uuid(), id: util.uuid(),
height, height,
width, width,
resolution_type: "1k", resolution_type: resolutionType,
}, },
}, },
history_option: { history_option: {
@ -312,6 +318,8 @@ export async function uploadImages(
refreshToken: string refreshToken: string
) { ) {
const model = getModel(_model); 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}`); logger.info(`使用模型: ${_model} 映射模型: ${model} ${width}x${height} 精细度: ${sampleStrength}`);
const { totalCredit } = await getCredit(refreshToken); const { totalCredit } = await getCredit(refreshToken);
@ -353,7 +361,7 @@ export async function uploadImages(
id: util.uuid(), id: util.uuid(),
min_version: DRAFT_VERSION, min_version: DRAFT_VERSION,
is_from_tsn: true, is_from_tsn: true,
version: DRAFT_VERSION, version: currentVersion,
main_component_id: componentId, main_component_id: componentId,
component_list: [ component_list: [
{ {