Method StableAudioSession

src/models/stable_audio/session.cpp:184–211

}  // namespace

StableAudioSession::StableAudioSession(
    runtime::TaskSpec task,
    runtime::SessionOptions options,
    std::shared_ptr<const StableAudioAssets> assets)
    : runtime::RuntimeSessionBase(options),
      task_(task),
      assets_(require_assets(std::move(assets))) {
    max_batch_ = runtime::parse_int_option(options.options, {"stable_audio.max_batch"}).value_or(max_batch_);
    if (max_batch_ <= 0) {
        throw std::runtime_error("stable_audio.max_batch must be positive");
    }
    if (const auto it = this->options().options.find("stable_audio.weight_type"); it != this->options().options.end()) {
        weight_storage_type_ = engine::assets::parse_tensor_storage_type(it->second);
        validate_weight_storage(weight_storage_type_, "stable_audio.weight_type");
    }
    if (const auto mem_saver = runtime::find_option(this->options().options, {"stable_audio.mem_saver"})) {
        mem_saver_ = runtime::parse_bool_option(*mem_saver, "stable_audio.mem_saver");
    }
    for (const auto & [key, value] : this->options().options) {
        if (key == "stable_audio.max_batch" || key == "stable_audio.weight_type" || key == "stable_audio.mem_saver") {
            (void) value;
            continue;
        }
        if (key.rfind("stable_audio.", 0) == 0) {
            throw std::runtime_error("unknown Stable Audio session option: " + key);
        }
    }
}

StableAudioSession::~StableAudioSession() = default;