Skip to content

Commit 2f3308d

Browse files
committed
allow settings and dict for videoInference, add new settings params, update bytedance providerSettings
1 parent d2e27a1 commit 2f3308d

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

runware/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1990,6 +1990,7 @@ def _buildVideoRequest(self, requestVideo: IVideoInference) -> Dict[str, Any]:
19901990
self._addOptionalField(request_object, requestVideo.inputs)
19911991
self._addProviderSettings(request_object, requestVideo)
19921992
self._addOptionalField(request_object, requestVideo.safety)
1993+
self._addOptionalField(request_object, requestVideo.settings)
19931994
self._addOptionalField(request_object, requestVideo.advancedFeatures)
19941995
self._addOptionalField(request_object, requestVideo.acceleratorOptions)
19951996

runware/types.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -752,6 +752,10 @@ class ISettings(SerializableMixin):
752752
turbo: Optional[bool] = None
753753
lyrics: Optional[str] = None
754754
guidanceType: Optional[str] = None
755+
# Video
756+
draft: Optional[bool] = None
757+
save_audio: Optional[bool] = None
758+
promptUpsampling: Optional[bool] = None
755759

756760
@property
757761
def request_key(self) -> str:
@@ -1217,6 +1221,7 @@ class IBytedanceProviderSettings(BaseProviderSettings):
12171221
fastMode: Optional[bool] = None # When enabled, speeds up generation by sacrificing some effects. Default: false. RTF: 25-28 (fast) vs 35 (normal)
12181222
audio: Optional[bool] = None
12191223
draft: Optional[bool] = None
1224+
optimizePromptMode: Optional[str] = None
12201225

12211226
@property
12221227
def provider_key(self) -> str:
@@ -1425,9 +1430,14 @@ class IVideoInference:
14251430
advancedFeatures: Optional[IVideoAdvancedFeatures] = None
14261431
acceleratorOptions: Optional[IAcceleratorOptions] = None
14271432
inputs: Optional[IVideoInputs] = None
1433+
settings: Optional[Union[ISettings, Dict[str, Any]]] = None
14281434
skipResponse: Optional[bool] = False
14291435
resolution: Optional[str] = None
14301436

1437+
def __post_init__(self):
1438+
if self.settings is not None and isinstance(self.settings, dict):
1439+
self.settings = ISettings(**self.settings)
1440+
14311441

14321442
I3dOutputFormat = Literal["GLB", "PLY"]
14331443

0 commit comments

Comments
 (0)