diff --git a/app/utils/model.ts b/app/utils/model.ts index 2681882ef..55a5ee0d6 100644 --- a/app/utils/model.ts +++ b/app/utils/model.ts @@ -71,10 +71,14 @@ export function collectModelTable( } // 2. if model not exists, create new model with available value if (count === 0) { - const [customModelName, customProviderName] = name.split("@"); + let [customModelName, customProviderName] = name.split("@"); const provider = customProvider( customProviderName || customModelName, ); + // swap name and displayName for bytedance + if (displayName && provider.providerName == "ByteDance") { + [customModelName, displayName] = [displayName, customModelName]; + } modelTable[`${customModelName}@${provider?.id}`] = { name: customModelName, displayName: displayName || customModelName,