multi transition first and last image saving

This commit is contained in:
Johannes Stelzer 2023-01-19 11:00:35 +01:00
parent 8d7bc4b49e
commit 56f68bd965
1 changed files with 7 additions and 1 deletions

View File

@ -109,6 +109,8 @@ class LatentBlending():
self.branch1_insertion_completed = False
self.set_guidance_scale(guidance_scale)
self.init_mode()
self.multi_transition_img_first = None
self.multi_transition_img_last = None
def init_mode(self):
@ -584,11 +586,15 @@ class LatentBlending():
list_imgs = self.run_transition(recycle_img1=recycle_img1, fixed_seeds=local_seeds)
list_imgs_interp = add_frames_linear_interp(list_imgs, fps, duration_single_trans)
if i==0:
self.multi_transition_img_first = list_imgs[0]
# Save movie frame
for img in list_imgs_interp:
ms.write_frame(img)
ms.finalize()
self.multi_transition_img_last = list_imgs[-1]
print("run_multi_transition: All completed.")