LocalDiffusionImpl

Implements LocalDiffusion behavior in the SDAI ONNX local diffusion feature layer.

Dmitriy Moroz

Constructors

constructor(uNet: UNet, tokenizer: LocalDiffusionTextTokenizer, ortEnvironmentProvider: OrtEnvironmentProvider)

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

private val statusFlow: MutableSharedFlow<LocalDiffusionStatus>

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

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

private val uNet: UNet

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

open suspend override fun interrupt()

Performs the SDAI side effect handled by interrupt.

Performs the SDAI side effect handled by interruptGeneration.

open override fun observeStatus(): SharedFlow<LocalDiffusionStatus>

Loads SDAI data through observeStatus.

open suspend override fun process(payload: TextToImagePayload): Bitmap

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