VaeDecoder

internal class VaeDecoder(val ortEnvironmentProvider: OrtEnvironmentProvider, val fileProviderDescriptor: FileProviderDescriptor, val localModelIdProvider: LocalModelIdProvider, val preferenceManager: PreferenceManager, val deviceId: Int)(source)

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

Dmitriy Moroz

Constructors

constructor(ortEnvironmentProvider: OrtEnvironmentProvider, fileProviderDescriptor: FileProviderDescriptor, localModelIdProvider: LocalModelIdProvider, preferenceManager: PreferenceManager, deviceId: Int)

private val deviceId: Int

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

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

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

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

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

private var session: OrtSession?

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

private fun clamp(value: Double, min: Double = 0.0, max: Double = 1.0): Double

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

fun close()

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

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

fun decode(input: Map<String?, OnnxTensor?>?): Any

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

private fun initialize()

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