feat: vidu video add starEnd and reference gen video
This commit is contained in:
@@ -11,8 +11,10 @@ const (
|
|||||||
SunoActionMusic = "MUSIC"
|
SunoActionMusic = "MUSIC"
|
||||||
SunoActionLyrics = "LYRICS"
|
SunoActionLyrics = "LYRICS"
|
||||||
|
|
||||||
TaskActionGenerate = "generate"
|
TaskActionGenerate = "generate"
|
||||||
TaskActionTextGenerate = "textGenerate"
|
TaskActionTextGenerate = "textGenerate"
|
||||||
|
TaskActionFirstTailGenerate = "firstTailGenerate"
|
||||||
|
TaskActionReferenceGenerate = "referenceGenerate"
|
||||||
)
|
)
|
||||||
|
|
||||||
var SunoModel2Action = map[string]string{
|
var SunoModel2Action = map[string]string{
|
||||||
|
|||||||
@@ -111,6 +111,10 @@ func (a *TaskAdaptor) BuildRequestURL(info *relaycommon.RelayInfo) (string, erro
|
|||||||
switch info.Action {
|
switch info.Action {
|
||||||
case constant.TaskActionGenerate:
|
case constant.TaskActionGenerate:
|
||||||
path = "/img2video"
|
path = "/img2video"
|
||||||
|
case constant.TaskActionFirstTailGenerate:
|
||||||
|
path = "/start-end2video"
|
||||||
|
case constant.TaskActionReferenceGenerate:
|
||||||
|
path = "/reference2video"
|
||||||
default:
|
default:
|
||||||
path = "/text2video"
|
path = "/text2video"
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -81,6 +81,14 @@ func ValidateBasicTaskRequest(c *gin.Context, info *RelayInfo, action string) *d
|
|||||||
|
|
||||||
if req.HasImage() {
|
if req.HasImage() {
|
||||||
action = constant.TaskActionGenerate
|
action = constant.TaskActionGenerate
|
||||||
|
if info.ChannelType == constant.ChannelTypeVidu {
|
||||||
|
// vidu 增加 首尾帧生视频和参考图生视频
|
||||||
|
if len(req.Images) == 2 {
|
||||||
|
action = constant.TaskActionFirstTailGenerate
|
||||||
|
} else if len(req.Images) > 2 {
|
||||||
|
action = constant.TaskActionReferenceGenerate
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
storeTaskRequest(c, info, action, req)
|
storeTaskRequest(c, info, action, req)
|
||||||
|
|||||||
Reference in New Issue
Block a user