48 lines
3.3 KiB
Python
48 lines
3.3 KiB
Python
from __future__ import annotations
|
|
|
|
import flatbuffers
|
|
import numpy as np
|
|
|
|
import flatbuffers
|
|
import typing
|
|
from ControlInputType import ControlInputType
|
|
from ControlMode import ControlMode
|
|
|
|
uoffset: typing.TypeAlias = flatbuffers.number_types.UOffsetTFlags.py_type
|
|
|
|
class Control(object):
|
|
@classmethod
|
|
def GetRootAs(cls, buf: bytes, offset: int) -> Control: ...
|
|
@classmethod
|
|
def GetRootAsControl(cls, buf: bytes, offset: int) -> Control: ...
|
|
def Init(self, buf: bytes, pos: int) -> None: ...
|
|
def File(self) -> str | None: ...
|
|
def Weight(self) -> float: ...
|
|
def GuidanceStart(self) -> float: ...
|
|
def GuidanceEnd(self) -> float: ...
|
|
def NoPrompt(self) -> bool: ...
|
|
def GlobalAveragePooling(self) -> bool: ...
|
|
def DownSamplingRate(self) -> float: ...
|
|
def ControlMode(self) -> typing.Literal[ControlMode.Balanced, ControlMode.Prompt, ControlMode.Control]: ...
|
|
def TargetBlocks(self, i: int) -> str: ...
|
|
def TargetBlocksLength(self) -> int: ...
|
|
def TargetBlocksIsNone(self) -> bool: ...
|
|
def InputOverride(self) -> typing.Literal[ControlInputType.Unspecified, ControlInputType.Custom, ControlInputType.Depth, ControlInputType.Canny, ControlInputType.Scribble, ControlInputType.Pose, ControlInputType.Normalbae, ControlInputType.Color, ControlInputType.Lineart, ControlInputType.Softedge, ControlInputType.Seg, ControlInputType.Inpaint, ControlInputType.Ip2p, ControlInputType.Shuffle, ControlInputType.Mlsd, ControlInputType.Tile, ControlInputType.Blur, ControlInputType.Lowquality, ControlInputType.Gray]: ...
|
|
def ControlStart(builder: flatbuffers.Builder) -> None: ...
|
|
def Start(builder: flatbuffers.Builder) -> None: ...
|
|
def ControlAddFile(builder: flatbuffers.Builder, file: uoffset) -> None: ...
|
|
def ControlAddWeight(builder: flatbuffers.Builder, weight: float) -> None: ...
|
|
def ControlAddGuidanceStart(builder: flatbuffers.Builder, guidanceStart: float) -> None: ...
|
|
def ControlAddGuidanceEnd(builder: flatbuffers.Builder, guidanceEnd: float) -> None: ...
|
|
def ControlAddNoPrompt(builder: flatbuffers.Builder, noPrompt: bool) -> None: ...
|
|
def ControlAddGlobalAveragePooling(builder: flatbuffers.Builder, globalAveragePooling: bool) -> None: ...
|
|
def ControlAddDownSamplingRate(builder: flatbuffers.Builder, downSamplingRate: float) -> None: ...
|
|
def ControlAddControlMode(builder: flatbuffers.Builder, controlMode: typing.Literal[ControlMode.Balanced, ControlMode.Prompt, ControlMode.Control]) -> None: ...
|
|
def ControlAddTargetBlocks(builder: flatbuffers.Builder, targetBlocks: uoffset) -> None: ...
|
|
def ControlStartTargetBlocksVector(builder: flatbuffers.Builder, num_elems: int) -> uoffset: ...
|
|
def StartTargetBlocksVector(builder: flatbuffers.Builder, num_elems: int) -> uoffset: ...
|
|
def ControlAddInputOverride(builder: flatbuffers.Builder, inputOverride: typing.Literal[ControlInputType.Unspecified, ControlInputType.Custom, ControlInputType.Depth, ControlInputType.Canny, ControlInputType.Scribble, ControlInputType.Pose, ControlInputType.Normalbae, ControlInputType.Color, ControlInputType.Lineart, ControlInputType.Softedge, ControlInputType.Seg, ControlInputType.Inpaint, ControlInputType.Ip2p, ControlInputType.Shuffle, ControlInputType.Mlsd, ControlInputType.Tile, ControlInputType.Blur, ControlInputType.Lowquality, ControlInputType.Gray]) -> None: ...
|
|
def ControlEnd(builder: flatbuffers.Builder) -> uoffset: ...
|
|
def End(builder: flatbuffers.Builder) -> uoffset: ...
|
|
|