Просмотр исходного кода

Merge pull request #2168 from feitianbubu/pr/fix-jimeng-1080p-image

fix: trim suffix p for jimeng image model
Seefs 1 месяц назад
Родитель
Сommit
2fd33ea294
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      relay/channel/task/jimeng/adaptor.go

+ 2 - 2
relay/channel/task/jimeng/adaptor.go

@@ -411,10 +411,10 @@ func (a *TaskAdaptor) convertToRequestPayload(req *relaycommon.TaskSubmitReq) (*
 			r.ReqKey = "jimeng_ti2v_v30_pro"
 		} else if len(req.Images) > 1 {
 			// 多张图片:首尾帧生成
-			r.ReqKey = strings.Replace(r.ReqKey, "jimeng_v30", "jimeng_i2v_first_tail_v30", 1)
+			r.ReqKey = strings.TrimSuffix(strings.Replace(r.ReqKey, "jimeng_v30", "jimeng_i2v_first_tail_v30", 1), "p")
 		} else if len(req.Images) == 1 {
 			// 单张图片:图生视频
-			r.ReqKey = strings.Replace(r.ReqKey, "jimeng_v30", "jimeng_i2v_first_v30", 1)
+			r.ReqKey = strings.TrimSuffix(strings.Replace(r.ReqKey, "jimeng_v30", "jimeng_i2v_first_v30", 1), "p")
 		} else {
 			// 无图片:文生视频
 			r.ReqKey = strings.Replace(r.ReqKey, "jimeng_v30", "jimeng_t2v_v30", 1)