Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
51 changes: 5 additions & 46 deletions runware/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -2050,54 +2050,13 @@ class ISourcePosition(SerializableMixin):
height: Optional[float] = None


@dataclass
class IVideoEditPoseControl(SerializableMixin):
enabled: Optional[bool] = None
strength: Optional[str] = None


@dataclass
class IVideoEditDepthControl(SerializableMixin):
enabled: Optional[bool] = None
blur: Optional[float] = None


@dataclass
class IVideoEditNormalsControl(SerializableMixin):
enabled: Optional[bool] = None
augmentation: Optional[float] = None


@dataclass
class IVideoEditTrajectoryControl(SerializableMixin):
enabled: Optional[bool] = None
sparsity: Optional[float] = None


@dataclass
class IVideoEditFaceControl(SerializableMixin):
enabled: Optional[bool] = None


@dataclass
class IVideoEditSettingsControls(SerializableMixin):
poseStrength: Optional[Union[IVideoEditPoseControl, Dict[str, Any]]] = None
depthBlur: Optional[Union[IVideoEditDepthControl, Dict[str, Any]]] = None
normalsAugmentation: Optional[Union[IVideoEditNormalsControl, Dict[str, Any]]] = None
trajectorySparsity: Optional[Union[IVideoEditTrajectoryControl, Dict[str, Any]]] = None
face: Optional[Union[IVideoEditFaceControl, Dict[str, Any]]] = None

def __post_init__(self):
if isinstance(self.poseStrength, dict):
self.poseStrength = IVideoEditPoseControl(**self.poseStrength)
if isinstance(self.depthBlur, dict):
self.depthBlur = IVideoEditDepthControl(**self.depthBlur)
if isinstance(self.normalsAugmentation, dict):
self.normalsAugmentation = IVideoEditNormalsControl(**self.normalsAugmentation)
if isinstance(self.trajectorySparsity, dict):
self.trajectorySparsity = IVideoEditTrajectoryControl(**self.trajectorySparsity)
if isinstance(self.face, dict):
self.face = IVideoEditFaceControl(**self.face)
poseStrength: Optional[str] = None
depthBlur: Optional[float] = None
Comment thread
Sirsho1997 marked this conversation as resolved.
normalsAugmentation: Optional[float] = None
trajectorySparsity: Optional[float] = None
face: Optional[bool] = None
Comment thread
Sirsho1997 marked this conversation as resolved.


@dataclass
Expand Down