(
self,
cp_size=0,
*args,
**kwargs,
)
| 551 | |
| 552 | class VideoAutoencoderInferenceWrapper(VideoAutoencodingEngine): |
| 553 | def __init__( |
| 554 | self, |
| 555 | cp_size=0, |
| 556 | *args, |
| 557 | **kwargs, |
| 558 | ): |
| 559 | self.cp_size = cp_size |
| 560 | return super().__init__(*args, **kwargs) |
| 561 | |
| 562 | def encode( |
| 563 | self, |