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

LocalDiffusionConfig

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)(source)

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

Author

Dmitriy Moroz

Constructors

LocalDiffusionConfig

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

Properties

algorithmType

val algorithmType: String

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

betaEnd

val betaEnd: Float

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

betaSchedule

val betaSchedule: String

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

betaStart

val betaStart: Float

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

clipSample

val clipSample: Boolean

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

clipSampleRange

val clipSampleRange: Float

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

dynamicThresholdingRatio

val dynamicThresholdingRatio: Float

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

lowerOrderFinal

val lowerOrderFinal: Boolean

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

predictionType

val predictionType: String

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

sampleMaxValue

val sampleMaxValue: Float

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

solverOrder

val solverOrder: Int

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

solverType

val solverType: String

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

thresholding

val thresholding: Boolean

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

trainedBetas

val trainedBetas: List<Float>

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