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

image 4.x

Browse Source
This commit is contained in:
jonathang4 2025-12-13 09:54:25 +08:00
parent 6654a0e076
commit e54a9b6f70
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/api/routes/images.ts
View File

@ -75,7 +75,7 @@ export default {
// 随机挑选一个refresh_token
const token = _.sample(tokens);
const {
// model,
model,
task_id,
prompt,
// negative_prompt: negativePrompt,
@ -88,7 +88,7 @@ export default {
if (USE_DATABASE_MODE) {
// 使用新的数据库方法
await DatabaseGenerationService.generateImagesV2('jimeng-3.0', task_id, prompt, {
await DatabaseGenerationService.generateImagesV2(model || 'jimeng-3.0', task_id, prompt, {
width,
height,
sampleStrength: 0.5,
@ -97,7 +97,7 @@ export default {
}, token);
} else {
// 使用原有方法（不等待结果）
generateImages('jimeng-3.0', task_id, prompt, {
generateImages(model || 'jimeng-3.0', task_id, prompt, {
width,
height,
sampleStrength: 0.5,

12
src/lib/services/TaskPollingService.ts
View File

@ -771,6 +771,9 @@ export class TaskPollingService {
// 映射模型
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",
@ -780,6 +783,11 @@ export class TaskPollingService {
"jimeng-xl-pro": "text2img_xl_sft",
};
const mappedModel = MODEL_MAP[model] || MODEL_MAP["jimeng-3.0"];
// 4.x 模型特殊参数
const isV4 = model.startsWith('jimeng-4');
const currentVersion = isV4 ? '3.3.7' : "3.0.2";
const resolutionType = isV4 ? '2k' : '1k';
const componentId = this.generateUUID();
const submitId = this.generateUUID();
@ -818,7 +826,7 @@ export class TaskPollingService {
id: this.generateUUID(),
min_version: "3.0.2",
is_from_tsn: true,
version: "3.0.2",
version: currentVersion,
main_component_id: componentId,
component_list: [
{
@ -847,7 +855,7 @@ export class TaskPollingService {
id: this.generateUUID(),
height,
width,
resolution_type: "1k",
resolution_type: resolutionType,
},
},
history_option: {