50 lines
1.9 KiB
Go
50 lines
1.9 KiB
Go
package public
|
|
|
|
// ModelType 模型类型常量
|
|
const (
|
|
ModelTypeInference = 100 // 推理模型
|
|
|
|
ModelTypeImage = 200 // 图片模型
|
|
ImageSubTypeTextToImage = 201 // 图片模型-文生图
|
|
ImageSubTypeImageToImage = 202 // 图片模型-图生图
|
|
ImageSubTypeImageEdit = 203 // 图片模型-图片编辑
|
|
ImageSubTypeImageVariation = 204 // 图片模型-图片变体
|
|
|
|
ModelTypeAudio = 300 // 音频模型
|
|
AudioSubTypeTextToSpeech = 301 // 音频模型-文生音
|
|
AudioSubTypeSpeechToText = 302 // 音频模型-音生文
|
|
AudioSubTypeSpeechToSpeech = 303 // 音频模型-音生音
|
|
|
|
ModelTypeVector = 400 // 向量化模型
|
|
VectorSubTypeEmbedding = 401 // 向量化模型-文本嵌入
|
|
VectorSubTypeRerank = 402 // 向量化模型-重排序
|
|
|
|
ModelTypeOmni = 500 // 全模态模型
|
|
OmniSubTypeTextImageAudio = 501 // 全模态模型-文图音
|
|
OmniSubTypeVision = 502 // 全模态模型-视觉理解
|
|
)
|
|
|
|
// ModelTypeName 模型类型名称映射
|
|
var ModelTypeName = map[int]string{
|
|
ModelTypeInference: "推理模型",
|
|
|
|
ModelTypeImage: "图片模型",
|
|
ImageSubTypeTextToImage: "图片模型-文生图",
|
|
ImageSubTypeImageToImage: "图片模型-图生图",
|
|
ImageSubTypeImageEdit: "图片模型-图片编辑",
|
|
ImageSubTypeImageVariation: "图片模型-图片变体",
|
|
|
|
ModelTypeAudio: "音频模型",
|
|
AudioSubTypeTextToSpeech: "音频模型-文生音",
|
|
AudioSubTypeSpeechToText: "音频模型-音生文",
|
|
AudioSubTypeSpeechToSpeech: "音频模型-音生音",
|
|
|
|
ModelTypeVector: "向量化模型",
|
|
VectorSubTypeEmbedding: "向量化模型-文本嵌入",
|
|
VectorSubTypeRerank: "向量化模型-重排序",
|
|
|
|
ModelTypeOmni: "全模态模型",
|
|
OmniSubTypeTextImageAudio: "全模态模型-文图音",
|
|
OmniSubTypeVision: "全模态模型-视觉理解",
|
|
}
|