| 24 | } |
| 25 | |
| 26 | engine::runtime::TaskRequest make_request( |
| 27 | const std::string & prompt, |
| 28 | const std::string & lyrics, |
| 29 | float duration_seconds, |
| 30 | int64_t num_inference_steps, |
| 31 | uint32_t seed) { |
| 32 | engine::runtime::TaskRequest request; |
| 33 | request.text_input = engine::runtime::Transcript{prompt, "en"}; |
| 34 | request.options["thinking"] = "true"; |
| 35 | request.options["lyrics"] = lyrics; |
| 36 | request.options["duration_seconds"] = std::to_string(duration_seconds); |
| 37 | request.options["num_inference_steps"] = std::to_string(num_inference_steps); |
| 38 | request.options["seed"] = std::to_string(seed); |
| 39 | return request; |
| 40 | } |
| 41 | |
| 42 | } // namespace |
| 43 | |