Method diffusion_weights

src/models/ace_step/dit_weights_runtime.cpp:329–331

}

std::shared_ptr<const AceStepDiffusionWeights> AceStepDitWeightsRuntime::diffusion_weights() const noexcept {
    return diffusion_weights_;
}

}  // namespace engine::models::ace_step