LocalDiffusionScheduler
Defines the LocalDiffusionScheduler contract for the SDAI ONNX local diffusion feature layer.
Author
Dmitriy Moroz
Inheritors
Properties
Functions
Link copied to clipboard
abstract fun scaleModelInput(sample: LocalDiffusionTensor<*>, stepIndex: Int): LocalDiffusionTensor<*>
Executes the scaleModelInput step in the SDAI ONNX local diffusion feature layer.
Link copied to clipboard
Executes the setTimeSteps step in the SDAI ONNX local diffusion feature layer.
Link copied to clipboard
abstract fun step(modelOutput: LocalDiffusionTensor<*>, stepIndex: Int, sample: LocalDiffusionTensor<*>): LocalDiffusionTensor<*>
Executes the step step in the SDAI ONNX local diffusion feature layer.