Skip to content

Commit fdf98dd

Browse files
committed
Added support for Runway Aleph 2
1 parent 15de442 commit fdf98dd

2 files changed

Lines changed: 10 additions & 1 deletion

File tree

runware/base.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2415,6 +2415,9 @@ async def _requestVideo(self, requestVideo: "IVideoInference") -> "Union[List[IV
24152415
processed_reference_videos.append(await self._process_media(item))
24162416
inputs.referenceVideos = processed_reference_videos
24172417

2418+
if inputs.video:
2419+
inputs.video = await self._process_media(inputs.video)
2420+
24182421
if inputs.frameImages:
24192422
inputs.frameImages = await self._process_media_list(
24202423
inputs.frameImages,

runware/types.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1170,6 +1170,7 @@ def __post_init__(self):
11701170
class IInputFrame(SerializableMixin):
11711171
image: Union[str, File]
11721172
frame: Optional[Union[Literal["first", "last"], int]] = None
1173+
timestamp: Optional[float] = None
11731174

11741175

11751176
@dataclass
@@ -1283,7 +1284,7 @@ class IVideoInputs(SerializableMixin):
12831284
image: Optional[Union[str, File]] = None
12841285
images: Optional[List[Union[str, File]]] = None
12851286
frames: Optional[List[IInputFrame]] = None
1286-
frameImages: Optional[List[IInputFrame]] = None
1287+
frameImages: Optional[List[Union[str, File, IInputFrame, Dict[str, Any]]]] = None
12871288
referenceImages: Optional[List[Union[str, File, IVideoReferenceImage]]] = None
12881289
referenceVideos: Optional[List[Union[str, IVideoReferenceVideo]]] = None
12891290
referenceAudios: Optional[List[str]] = None
@@ -1354,6 +1355,11 @@ def __post_init__(self):
13541355
IElements(**item) if isinstance(item, dict) else item
13551356
for item in self.elements
13561357
]
1358+
if self.frameImages:
1359+
self.frameImages = [
1360+
IInputFrame(**item) if isinstance(item, dict) else item
1361+
for item in self.frameImages
1362+
]
13571363

13581364
@property
13591365
def request_key(self) -> str:

0 commit comments

Comments
 (0)