Function parse_stable_audio_request

src/models/stable_audio/request.cpp:138–205

}  // namespace

StableAudioRequest parse_stable_audio_request(const runtime::TaskRequest & request) {
    StableAudioRequest out;
    if (const auto value = runtime::parse_int_option(request.options, {"batch_size"})) {
        if (*value <= 0) {
            throw std::runtime_error("Stable Audio batch_size must be positive");
        }
        out.batch_size = *value;
    }
    out.prompts = prompts_from_request(request, out.batch_size);
    out.negative_prompts = negative_prompts_from_request(request, out.batch_size);
    out.durations_seconds = durations_from_request(request, out.batch_size);
    if (const auto value = runtime::parse_int_option(request.options, {"num_inference_steps"})) {
        if (*value <= 0) {
            throw std::runtime_error("Stable Audio num_inference_steps must be positive");
        }
        out.num_inference_steps = *value;
    }
    if (const auto value = runtime::find_option(request.options, {"sampler"})) {
        if (*value != "pingpong" && *value != "euler") {
            throw std::runtime_error("Stable Audio sampler must be pingpong or euler");
        }
        out.sampler = *value;
    }
    if (const auto value = runtime::parse_finite_float_option(request.options, {"guidance_scale"})) {
        out.guidance_scale = *value;
    }
    if (const auto value = runtime::parse_finite_float_option(request.options, {"apg_scale"})) {
        out.apg_scale = *value;
    }
    if (const auto value = runtime::parse_u64_option(request.options, {"seed"})) {
        out.seed = *value;
        out.seed_specified = true;
    } else {
        out.seed = runtime::random_u64_seed();
    }
    if (const auto value = runtime::find_option(request.options, {"truncate_output_to_duration"})) {
        out.truncate_output_to_duration = runtime::parse_bool_option(*value, "truncate_output_to_duration");
    }
    if (const auto value = runtime::find_option(request.options, {"chunked_decode"})) {
        out.chunked_decode = runtime::parse_bool_option(*value, "chunked_decode");
    }
    if (const auto value = runtime::parse_finite_float_option(request.options, {"duration_padding_seconds"})) {
        if (*value < 0.0F) {
            throw std::runtime_error("Stable Audio duration_padding_seconds must be non-negative");
        }
        out.duration_padding_seconds = *value;
    }
    if (const auto value = runtime::parse_finite_float_option(request.options, {"init_noise_level"})) {
        if (*value < 0.0F || *value > 1.0F) {
            throw std::runtime_error("Stable Audio init_noise_level must be in [0, 1]");
        }
        out.init_noise_level = *value;
    }
    const auto audio_input_kind = runtime::find_option(request.options, {"audio_input_kind"});
    if (audio_input_kind.has_value() && *audio_input_kind != "init_audio" && *audio_input_kind != "inpaint_audio") {
        throw std::runtime_error("Stable Audio audio_input_kind must be init_audio or inpaint_audio");
    }
    if (request.audio_input.has_value() && (!audio_input_kind.has_value() || *audio_input_kind == "init_audio")) {