feature/onnx/com.shifthackz.aisdv1.feature.onnx.ai.scheduler/EulerAncestralDiscreteLocalDiffusionScheduler

EulerAncestralDiscreteLocalDiffusionScheduler

internal class EulerAncestralDiscreteLocalDiffusionScheduler(val config: LocalDiffusionConfig = LocalDiffusionConfig()) : LocalDiffusionScheduler, KoinComponent(source)

Coordinates EulerAncestralDiscreteLocalDiffusionScheduler behavior in the SDAI ONNX local diffusion feature layer.

Author

Dmitriy Moroz

Constructors

EulerAncestralDiscreteLocalDiffusionScheduler

constructor(config: LocalDiffusionConfig = LocalDiffusionConfig())

Properties

alphas

private val alphas: MutableList<Float>

alphasCumpRod

private val alphasCumpRod: MutableList<Float>

betas

private val betas: MutableList<Float>

config

private val config: LocalDiffusionConfig

Exposes the config value used by the SDAI ONNX local diffusion feature layer.

initNoiseSigma

open override var initNoiseSigma: Double

Exposes the initNoiseSigma value used by the SDAI ONNX local diffusion feature layer.

isScaleInputCalled

private var isScaleInputCalled: Boolean

numInferenceSteps

private var numInferenceSteps: Int

numTrainTimeSteps

private val numTrainTimeSteps: Int = 1000

ortEnvironmentProvider

private val ortEnvironmentProvider: OrtEnvironmentProvider

random

private val random: Random

sigmas

private val sigmas: MutableList<Float>

timeSteps

private val timeSteps: MutableList<Int>

Functions

alphaBar

private fun alphaBar(timeStep: Double): Double

betasForAlphaBar

private fun betasForAlphaBar(numDiffusionTimeSteps: Int = 1000, maxBeta: Float = 0.999f): List<Float>

getKoin

open fun getKoin(): Koin

scaleModelInput

open override fun scaleModelInput(sample: LocalDiffusionTensor<*>, stepIndex: Int): LocalDiffusionTensor<*>

Executes the scaleModelInput step in the SDAI ONNX local diffusion feature layer.

setTimeSteps

open override fun setTimeSteps(numInferenceSteps: Int): IntArray

Executes the setTimeSteps step in the SDAI ONNX local diffusion feature layer.

step

open override fun step(modelOutput: LocalDiffusionTensor<*>, stepIndex: Int, sample: LocalDiffusionTensor<*>): LocalDiffusionTensor<*>

Executes the step step in the SDAI ONNX local diffusion feature layer.