diff --git a/src/views/ai/image/index/components/other/index.vue b/src/views/ai/image/index/components/other/index.vue index 375c15b6..861a1563 100644 --- a/src/views/ai/image/index/components/other/index.vue +++ b/src/views/ai/image/index/components/other/index.vue @@ -83,6 +83,7 @@ import { ImageHotWords, ImageModelVO, OtherPlatformEnum, + QianFanModels, TongYiWanXiangModels } from '@/views/ai/utils/constants' @@ -96,7 +97,7 @@ const prompt = ref('') // 提示词 const width = ref(512) // 图片宽度 const height = ref(512) // 图片高度 const otherPlatform = ref(AiPlatformEnum.TONG_YI) // 平台 -const models = ref(TongYiWanXiangModels) // 模型 +const models = ref(TongYiWanXiangModels) // 模型 TongYiWanXiangModels、QianFanModels const model = ref(models.value[0].key) // 模型 @@ -154,8 +155,10 @@ const settingValues = async (detail: ImageVO) => { /** 平台切换 */ const handlerPlatformChange = async (platform) => { // 切换平台,切换模型、风格 - if (AiPlatformEnum.YI_YAN === platform) { + if (AiPlatformEnum.TONG_YI === platform) { models.value = TongYiWanXiangModels + } else if (AiPlatformEnum.YI_YAN === platform) { + models.value = QianFanModels } else { models.value = [] } diff --git a/src/views/ai/utils/constants.ts b/src/views/ai/utils/constants.ts index 0a88775b..d33493e2 100644 --- a/src/views/ai/utils/constants.ts +++ b/src/views/ai/utils/constants.ts @@ -30,7 +30,7 @@ export const OtherPlatformEnum:ImageModelVO [] = [ }, { key: AiPlatformEnum.YI_YAN, - name: '百度图片' + name: '百度千帆' } ] @@ -259,6 +259,14 @@ export const TongYiWanXiangModels: ImageModelVO[] = [ } ] +export const QianFanModels: ImageModelVO[] = [ + { + key: 'sd_xl', + name: 'sd_xl' + } +] + + export const StableDiffusionClipGuidancePresets: ImageModelVO[] = [ { key: 'NONE',