(
vae, generated_video, refs_x, mask_strategy, loop_i, condition_frame_length, condition_frame_edit, is_latent=False
)
| 316 | |
| 317 | |
| 318 | def append_generated( |
| 319 | vae, generated_video, refs_x, mask_strategy, loop_i, condition_frame_length, condition_frame_edit, is_latent=False |
| 320 | ): |
| 321 | ref_x = vae.encode(generated_video) if not is_latent else generated_video |
| 322 | |
| 323 | for j, refs in enumerate(refs_x): |
| 324 | if refs is None: |
| 325 | refs_x[j] = [ref_x[j]] |
| 326 | else: |
| 327 | refs.append(ref_x[j]) |
| 328 | if mask_strategy[j] is None or mask_strategy[j] == "": |
| 329 | mask_strategy[j] = "" |
| 330 | else: |
| 331 | mask_strategy[j] += ";" |
| 332 | mask_strategy[ |
| 333 | j |
| 334 | ] += f"{loop_i},{len(refs)-1},-{condition_frame_length},0,{condition_frame_length},{condition_frame_edit}" |
| 335 | return refs_x, mask_strategy |
| 336 | |
| 337 | |
| 338 | def dframe_to_frame(num): |
no test coverage detected