Jelajahi Sumber

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

fix: trim suffix p for jimeng image model
Seefs 1 bulan lalu
induk
melakukan
2fd33ea294
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  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"
 			r.ReqKey = "jimeng_ti2v_v30_pro"
 		} else if len(req.Images) > 1 {
 		} 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 {
 		} 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 {
 		} else {
 			// 无图片:文生视频
 			// 无图片:文生视频
 			r.ReqKey = strings.Replace(r.ReqKey, "jimeng_v30", "jimeng_t2v_v30", 1)
 			r.ReqKey = strings.Replace(r.ReqKey, "jimeng_v30", "jimeng_t2v_v30", 1)