Method RoformerSession

src/models/roformer/session.cpp:119–156

        throw std::runtime_error("RoFormer config num_overlap must be positive");
    }
    chunk_size_ = config.chunk_size;
    const int64_t overlap = runtime::parse_positive_i64_option(
        RuntimeSessionBase::options().options,
        {config.family + ".num_overlap"},
        config.inference_num_overlap);
    if (overlap > chunk_size_) {
        throw std::runtime_error(config.family + ".num_overlap must not exceed chunk_size");
    }
    step_ = chunk_size_ / overlap;
    fade_size_ = chunk_size_ / 10;
    border_ = chunk_size_ - step_;
    if (step_ <= 0) {
        throw std::runtime_error("RoFormer chunk step must be positive");
    }
    chunk_window_ = engine::audio::make_linear_fade_window(chunk_size_, fade_size_);
    first_chunk_window_ = chunk_window_;
    std::fill(first_chunk_window_.begin(), first_chunk_window_.begin() + fade_size_, 1.0f);
    last_chunk_window_ = chunk_window_;
    std::fill(last_chunk_window_.end() - fade_size_, last_chunk_window_.end(), 1.0f);
    only_chunk_window_ = first_chunk_window_;
    std::fill(only_chunk_window_.end() - fade_size_, only_chunk_window_.end(), 1.0f);
    chunk_planar_work_.resize(static_cast<size_t>(config.channels * chunk_size_));
    assets_->tensor_source->release_storage();
}

RoformerSession::~RoformerSession() = default;

std::string RoformerSession::family() const {
    return assets_->config.family;
}

runtime::VoiceTaskKind RoformerSession::task_kind() const {
    return task_.task;
}

runtime::RunMode RoformerSession::run_mode() const {
    return task_.mode;
}

void RoformerSession::prepare(const runtime::SessionPreparationRequest & request) {
    if (!request.audio.has_value()) {