188 lines
14 KiB
Python
188 lines
14 KiB
Python
from __future__ import annotations
|
|
|
|
import flatbuffers
|
|
import numpy as np
|
|
|
|
import flatbuffers
|
|
import typing
|
|
from Control import Control
|
|
from LoRA import LoRA
|
|
from SamplerType import SamplerType
|
|
from SeedMode import SeedMode
|
|
|
|
uoffset: typing.TypeAlias = flatbuffers.number_types.UOffsetTFlags.py_type
|
|
|
|
class GenerationConfiguration(object):
|
|
@classmethod
|
|
def GetRootAs(cls, buf: bytes, offset: int) -> GenerationConfiguration: ...
|
|
@classmethod
|
|
def GetRootAsGenerationConfiguration(cls, buf: bytes, offset: int) -> GenerationConfiguration: ...
|
|
def Init(self, buf: bytes, pos: int) -> None: ...
|
|
def Id(self) -> int: ...
|
|
def StartWidth(self) -> int: ...
|
|
def StartHeight(self) -> int: ...
|
|
def Seed(self) -> int: ...
|
|
def Steps(self) -> int: ...
|
|
def GuidanceScale(self) -> float: ...
|
|
def Strength(self) -> float: ...
|
|
def Model(self) -> str | None: ...
|
|
def Sampler(self) -> typing.Literal[SamplerType.DPMPP2MKarras, SamplerType.EulerA, SamplerType.DDIM, SamplerType.PLMS, SamplerType.DPMPPSDEKarras, SamplerType.UniPC, SamplerType.LCM, SamplerType.EulerASubstep, SamplerType.DPMPPSDESubstep, SamplerType.TCD, SamplerType.EulerATrailing, SamplerType.DPMPPSDETrailing, SamplerType.DPMPP2MAYS, SamplerType.EulerAAYS, SamplerType.DPMPPSDEAYS, SamplerType.DPMPP2MTrailing, SamplerType.DDIMTrailing]: ...
|
|
def BatchCount(self) -> int: ...
|
|
def BatchSize(self) -> int: ...
|
|
def HiresFix(self) -> bool: ...
|
|
def HiresFixStartWidth(self) -> int: ...
|
|
def HiresFixStartHeight(self) -> int: ...
|
|
def HiresFixStrength(self) -> float: ...
|
|
def Upscaler(self) -> str | None: ...
|
|
def ImageGuidanceScale(self) -> float: ...
|
|
def SeedMode(self) -> typing.Literal[SeedMode.Legacy, SeedMode.TorchCpuCompatible, SeedMode.ScaleAlike, SeedMode.NvidiaGpuCompatible]: ...
|
|
def ClipSkip(self) -> int: ...
|
|
def Controls(self, i: int) -> Control | None: ...
|
|
def ControlsLength(self) -> int: ...
|
|
def ControlsIsNone(self) -> bool: ...
|
|
def Loras(self, i: int) -> LoRA | None: ...
|
|
def LorasLength(self) -> int: ...
|
|
def LorasIsNone(self) -> bool: ...
|
|
def MaskBlur(self) -> float: ...
|
|
def FaceRestoration(self) -> str | None: ...
|
|
def ClipWeight(self) -> float: ...
|
|
def NegativePromptForImagePrior(self) -> bool: ...
|
|
def ImagePriorSteps(self) -> int: ...
|
|
def RefinerModel(self) -> str | None: ...
|
|
def OriginalImageHeight(self) -> int: ...
|
|
def OriginalImageWidth(self) -> int: ...
|
|
def CropTop(self) -> int: ...
|
|
def CropLeft(self) -> int: ...
|
|
def TargetImageHeight(self) -> int: ...
|
|
def TargetImageWidth(self) -> int: ...
|
|
def AestheticScore(self) -> float: ...
|
|
def NegativeAestheticScore(self) -> float: ...
|
|
def ZeroNegativePrompt(self) -> bool: ...
|
|
def RefinerStart(self) -> float: ...
|
|
def NegativeOriginalImageHeight(self) -> int: ...
|
|
def NegativeOriginalImageWidth(self) -> int: ...
|
|
def Name(self) -> str | None: ...
|
|
def FpsId(self) -> int: ...
|
|
def MotionBucketId(self) -> int: ...
|
|
def CondAug(self) -> float: ...
|
|
def StartFrameCfg(self) -> float: ...
|
|
def NumFrames(self) -> int: ...
|
|
def MaskBlurOutset(self) -> int: ...
|
|
def Sharpness(self) -> float: ...
|
|
def Shift(self) -> float: ...
|
|
def Stage2Steps(self) -> int: ...
|
|
def Stage2Cfg(self) -> float: ...
|
|
def Stage2Shift(self) -> float: ...
|
|
def TiledDecoding(self) -> bool: ...
|
|
def DecodingTileWidth(self) -> int: ...
|
|
def DecodingTileHeight(self) -> int: ...
|
|
def DecodingTileOverlap(self) -> int: ...
|
|
def StochasticSamplingGamma(self) -> float: ...
|
|
def PreserveOriginalAfterInpaint(self) -> bool: ...
|
|
def TiledDiffusion(self) -> bool: ...
|
|
def DiffusionTileWidth(self) -> int: ...
|
|
def DiffusionTileHeight(self) -> int: ...
|
|
def DiffusionTileOverlap(self) -> int: ...
|
|
def UpscalerScaleFactor(self) -> int: ...
|
|
def T5TextEncoder(self) -> bool: ...
|
|
def SeparateClipL(self) -> bool: ...
|
|
def ClipLText(self) -> str | None: ...
|
|
def SeparateOpenClipG(self) -> bool: ...
|
|
def OpenClipGText(self) -> str | None: ...
|
|
def SpeedUpWithGuidanceEmbed(self) -> bool: ...
|
|
def GuidanceEmbed(self) -> float: ...
|
|
def ResolutionDependentShift(self) -> bool: ...
|
|
def TeaCacheStart(self) -> int: ...
|
|
def TeaCacheEnd(self) -> int: ...
|
|
def TeaCacheThreshold(self) -> float: ...
|
|
def TeaCache(self) -> bool: ...
|
|
def SeparateT5(self) -> bool: ...
|
|
def T5Text(self) -> str | None: ...
|
|
def TeaCacheMaxSkipSteps(self) -> int: ...
|
|
def GenerationConfigurationStart(builder: flatbuffers.Builder) -> None: ...
|
|
def Start(builder: flatbuffers.Builder) -> None: ...
|
|
def GenerationConfigurationAddId(builder: flatbuffers.Builder, id: int) -> None: ...
|
|
def GenerationConfigurationAddStartWidth(builder: flatbuffers.Builder, startWidth: int) -> None: ...
|
|
def GenerationConfigurationAddStartHeight(builder: flatbuffers.Builder, startHeight: int) -> None: ...
|
|
def GenerationConfigurationAddSeed(builder: flatbuffers.Builder, seed: int) -> None: ...
|
|
def GenerationConfigurationAddSteps(builder: flatbuffers.Builder, steps: int) -> None: ...
|
|
def GenerationConfigurationAddGuidanceScale(builder: flatbuffers.Builder, guidanceScale: float) -> None: ...
|
|
def GenerationConfigurationAddStrength(builder: flatbuffers.Builder, strength: float) -> None: ...
|
|
def GenerationConfigurationAddModel(builder: flatbuffers.Builder, model: uoffset) -> None: ...
|
|
def GenerationConfigurationAddSampler(builder: flatbuffers.Builder, sampler: typing.Literal[SamplerType.DPMPP2MKarras, SamplerType.EulerA, SamplerType.DDIM, SamplerType.PLMS, SamplerType.DPMPPSDEKarras, SamplerType.UniPC, SamplerType.LCM, SamplerType.EulerASubstep, SamplerType.DPMPPSDESubstep, SamplerType.TCD, SamplerType.EulerATrailing, SamplerType.DPMPPSDETrailing, SamplerType.DPMPP2MAYS, SamplerType.EulerAAYS, SamplerType.DPMPPSDEAYS, SamplerType.DPMPP2MTrailing, SamplerType.DDIMTrailing]) -> None: ...
|
|
def GenerationConfigurationAddBatchCount(builder: flatbuffers.Builder, batchCount: int) -> None: ...
|
|
def GenerationConfigurationAddBatchSize(builder: flatbuffers.Builder, batchSize: int) -> None: ...
|
|
def GenerationConfigurationAddHiresFix(builder: flatbuffers.Builder, hiresFix: bool) -> None: ...
|
|
def GenerationConfigurationAddHiresFixStartWidth(builder: flatbuffers.Builder, hiresFixStartWidth: int) -> None: ...
|
|
def GenerationConfigurationAddHiresFixStartHeight(builder: flatbuffers.Builder, hiresFixStartHeight: int) -> None: ...
|
|
def GenerationConfigurationAddHiresFixStrength(builder: flatbuffers.Builder, hiresFixStrength: float) -> None: ...
|
|
def GenerationConfigurationAddUpscaler(builder: flatbuffers.Builder, upscaler: uoffset) -> None: ...
|
|
def GenerationConfigurationAddImageGuidanceScale(builder: flatbuffers.Builder, imageGuidanceScale: float) -> None: ...
|
|
def GenerationConfigurationAddSeedMode(builder: flatbuffers.Builder, seedMode: typing.Literal[SeedMode.Legacy, SeedMode.TorchCpuCompatible, SeedMode.ScaleAlike, SeedMode.NvidiaGpuCompatible]) -> None: ...
|
|
def GenerationConfigurationAddClipSkip(builder: flatbuffers.Builder, clipSkip: int) -> None: ...
|
|
def GenerationConfigurationAddControls(builder: flatbuffers.Builder, controls: uoffset) -> None: ...
|
|
def GenerationConfigurationStartControlsVector(builder: flatbuffers.Builder, num_elems: int) -> uoffset: ...
|
|
def StartControlsVector(builder: flatbuffers.Builder, num_elems: int) -> uoffset: ...
|
|
def GenerationConfigurationAddLoras(builder: flatbuffers.Builder, loras: uoffset) -> None: ...
|
|
def GenerationConfigurationStartLorasVector(builder: flatbuffers.Builder, num_elems: int) -> uoffset: ...
|
|
def StartLorasVector(builder: flatbuffers.Builder, num_elems: int) -> uoffset: ...
|
|
def GenerationConfigurationAddMaskBlur(builder: flatbuffers.Builder, maskBlur: float) -> None: ...
|
|
def GenerationConfigurationAddFaceRestoration(builder: flatbuffers.Builder, faceRestoration: uoffset) -> None: ...
|
|
def GenerationConfigurationAddClipWeight(builder: flatbuffers.Builder, clipWeight: float) -> None: ...
|
|
def GenerationConfigurationAddNegativePromptForImagePrior(builder: flatbuffers.Builder, negativePromptForImagePrior: bool) -> None: ...
|
|
def GenerationConfigurationAddImagePriorSteps(builder: flatbuffers.Builder, imagePriorSteps: int) -> None: ...
|
|
def GenerationConfigurationAddRefinerModel(builder: flatbuffers.Builder, refinerModel: uoffset) -> None: ...
|
|
def GenerationConfigurationAddOriginalImageHeight(builder: flatbuffers.Builder, originalImageHeight: int) -> None: ...
|
|
def GenerationConfigurationAddOriginalImageWidth(builder: flatbuffers.Builder, originalImageWidth: int) -> None: ...
|
|
def GenerationConfigurationAddCropTop(builder: flatbuffers.Builder, cropTop: int) -> None: ...
|
|
def GenerationConfigurationAddCropLeft(builder: flatbuffers.Builder, cropLeft: int) -> None: ...
|
|
def GenerationConfigurationAddTargetImageHeight(builder: flatbuffers.Builder, targetImageHeight: int) -> None: ...
|
|
def GenerationConfigurationAddTargetImageWidth(builder: flatbuffers.Builder, targetImageWidth: int) -> None: ...
|
|
def GenerationConfigurationAddAestheticScore(builder: flatbuffers.Builder, aestheticScore: float) -> None: ...
|
|
def GenerationConfigurationAddNegativeAestheticScore(builder: flatbuffers.Builder, negativeAestheticScore: float) -> None: ...
|
|
def GenerationConfigurationAddZeroNegativePrompt(builder: flatbuffers.Builder, zeroNegativePrompt: bool) -> None: ...
|
|
def GenerationConfigurationAddRefinerStart(builder: flatbuffers.Builder, refinerStart: float) -> None: ...
|
|
def GenerationConfigurationAddNegativeOriginalImageHeight(builder: flatbuffers.Builder, negativeOriginalImageHeight: int) -> None: ...
|
|
def GenerationConfigurationAddNegativeOriginalImageWidth(builder: flatbuffers.Builder, negativeOriginalImageWidth: int) -> None: ...
|
|
def GenerationConfigurationAddName(builder: flatbuffers.Builder, name: uoffset) -> None: ...
|
|
def GenerationConfigurationAddFpsId(builder: flatbuffers.Builder, fpsId: int) -> None: ...
|
|
def GenerationConfigurationAddMotionBucketId(builder: flatbuffers.Builder, motionBucketId: int) -> None: ...
|
|
def GenerationConfigurationAddCondAug(builder: flatbuffers.Builder, condAug: float) -> None: ...
|
|
def GenerationConfigurationAddStartFrameCfg(builder: flatbuffers.Builder, startFrameCfg: float) -> None: ...
|
|
def GenerationConfigurationAddNumFrames(builder: flatbuffers.Builder, numFrames: int) -> None: ...
|
|
def GenerationConfigurationAddMaskBlurOutset(builder: flatbuffers.Builder, maskBlurOutset: int) -> None: ...
|
|
def GenerationConfigurationAddSharpness(builder: flatbuffers.Builder, sharpness: float) -> None: ...
|
|
def GenerationConfigurationAddShift(builder: flatbuffers.Builder, shift: float) -> None: ...
|
|
def GenerationConfigurationAddStage2Steps(builder: flatbuffers.Builder, stage2Steps: int) -> None: ...
|
|
def GenerationConfigurationAddStage2Cfg(builder: flatbuffers.Builder, stage2Cfg: float) -> None: ...
|
|
def GenerationConfigurationAddStage2Shift(builder: flatbuffers.Builder, stage2Shift: float) -> None: ...
|
|
def GenerationConfigurationAddTiledDecoding(builder: flatbuffers.Builder, tiledDecoding: bool) -> None: ...
|
|
def GenerationConfigurationAddDecodingTileWidth(builder: flatbuffers.Builder, decodingTileWidth: int) -> None: ...
|
|
def GenerationConfigurationAddDecodingTileHeight(builder: flatbuffers.Builder, decodingTileHeight: int) -> None: ...
|
|
def GenerationConfigurationAddDecodingTileOverlap(builder: flatbuffers.Builder, decodingTileOverlap: int) -> None: ...
|
|
def GenerationConfigurationAddStochasticSamplingGamma(builder: flatbuffers.Builder, stochasticSamplingGamma: float) -> None: ...
|
|
def GenerationConfigurationAddPreserveOriginalAfterInpaint(builder: flatbuffers.Builder, preserveOriginalAfterInpaint: bool) -> None: ...
|
|
def GenerationConfigurationAddTiledDiffusion(builder: flatbuffers.Builder, tiledDiffusion: bool) -> None: ...
|
|
def GenerationConfigurationAddDiffusionTileWidth(builder: flatbuffers.Builder, diffusionTileWidth: int) -> None: ...
|
|
def GenerationConfigurationAddDiffusionTileHeight(builder: flatbuffers.Builder, diffusionTileHeight: int) -> None: ...
|
|
def GenerationConfigurationAddDiffusionTileOverlap(builder: flatbuffers.Builder, diffusionTileOverlap: int) -> None: ...
|
|
def GenerationConfigurationAddUpscalerScaleFactor(builder: flatbuffers.Builder, upscalerScaleFactor: int) -> None: ...
|
|
def GenerationConfigurationAddT5TextEncoder(builder: flatbuffers.Builder, t5TextEncoder: bool) -> None: ...
|
|
def GenerationConfigurationAddSeparateClipL(builder: flatbuffers.Builder, separateClipL: bool) -> None: ...
|
|
def GenerationConfigurationAddClipLText(builder: flatbuffers.Builder, clipLText: uoffset) -> None: ...
|
|
def GenerationConfigurationAddSeparateOpenClipG(builder: flatbuffers.Builder, separateOpenClipG: bool) -> None: ...
|
|
def GenerationConfigurationAddOpenClipGText(builder: flatbuffers.Builder, openClipGText: uoffset) -> None: ...
|
|
def GenerationConfigurationAddSpeedUpWithGuidanceEmbed(builder: flatbuffers.Builder, speedUpWithGuidanceEmbed: bool) -> None: ...
|
|
def GenerationConfigurationAddGuidanceEmbed(builder: flatbuffers.Builder, guidanceEmbed: float) -> None: ...
|
|
def GenerationConfigurationAddResolutionDependentShift(builder: flatbuffers.Builder, resolutionDependentShift: bool) -> None: ...
|
|
def GenerationConfigurationAddTeaCacheStart(builder: flatbuffers.Builder, teaCacheStart: int) -> None: ...
|
|
def GenerationConfigurationAddTeaCacheEnd(builder: flatbuffers.Builder, teaCacheEnd: int) -> None: ...
|
|
def GenerationConfigurationAddTeaCacheThreshold(builder: flatbuffers.Builder, teaCacheThreshold: float) -> None: ...
|
|
def GenerationConfigurationAddTeaCache(builder: flatbuffers.Builder, teaCache: bool) -> None: ...
|
|
def GenerationConfigurationAddSeparateT5(builder: flatbuffers.Builder, separateT5: bool) -> None: ...
|
|
def GenerationConfigurationAddT5Text(builder: flatbuffers.Builder, t5Text: uoffset) -> None: ...
|
|
def GenerationConfigurationAddTeaCacheMaxSkipSteps(builder: flatbuffers.Builder, teaCacheMaxSkipSteps: int) -> None: ...
|
|
def GenerationConfigurationEnd(builder: flatbuffers.Builder) -> uoffset: ...
|
|
def End(builder: flatbuffers.Builder) -> uoffset: ...
|
|
|