feature/onnx/com.shifthackz.aisdv1.feature.onnx.entity

Package-level declarations

Types

typealias Array3D<T> = Array<Array<Array<T>>>

Aliases Array3D for SDAI ONNX local diffusion feature code.

internal data class LocalDiffusionConfig(val betaStart: Float = 8.5E-4f, val betaEnd: Float = 0.012f, val betaSchedule: String = BETA_SCHEDULER_SCALED_LINEAR, val trainedBetas: List<Float> = emptyList(), val solverOrder: Int = 2, val predictionType: String = PREDICTION_EPSILON, val thresholding: Boolean = false, val dynamicThresholdingRatio: Float = 0.995f, val sampleMaxValue: Float = 1.0f, val algorithmType: String = DPM_SOLVER_PP, val solverType: String = SOLVER_MIDPOINT, val lowerOrderFinal: Boolean = true, val clipSample: Boolean = false, val clipSampleRange: Float = 1.0f)

Carries LocalDiffusionConfig data through the SDAI ONNX local diffusion feature layer.

LocalDiffusionFlag

enum LocalDiffusionFlag(val value: Int) : Enum<LocalDiffusionFlag>

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

LocalDiffusionTensor

internal data class LocalDiffusionTensor<T>(var tensor: OnnxTensor, var buffer: T?, var shape: LongArray?)

Carries LocalDiffusionTensor data through the SDAI ONNX local diffusion feature layer.