Merge pull request #4798 from vladmandic/feat/rife-upgrade

Feat/rife upgrade
This commit is contained in:
Vladimir Mandic
2026-04-27 07:17:47 +02:00
committed by GitHub
17 changed files with 315 additions and 136 deletions
+2
View File
@@ -248,6 +248,7 @@ class AnimateDiffScript(scripts_manager.Script):
p.extra_generation_params['AnimateDiff'] = loaded_adapter
p.do_not_save_grid = True
p.ops.append('video')
p.video_interpolate = mp4_interpolate
p.task_args['generator'] = None
p.task_args['num_frames'] = frames
p.task_args['num_inference_steps'] = p.steps
@@ -267,3 +268,4 @@ class AnimateDiffScript(scripts_manager.Script):
if video_type != 'None':
log.debug(f'AnimateDiff video: type={video_type} duration={duration} loop={gif_loop} pad={mp4_pad} interpolate={mp4_interpolate}')
save_video(p, filename=None, images=processed.images, video_type=video_type, duration=duration, loop=gif_loop, pad=mp4_pad, interpolate=mp4_interpolate)
p.video_saved = True
+2
View File
@@ -57,6 +57,7 @@ class VGenI2VScript(scripts_manager.Script):
return None
if p.init_images is None or len(p.init_images) == 0:
return None
p.video_interpolate = mp4_interpolate
model = [m for m in MODELS if m['name'] == model_name][0]
repo_id = model['url']
log.debug(f'Image2Video: model={model_name} frames={num_frames}, video={video_type} duration={duration} loop={gif_loop} pad={mp4_pad} interpolate={mp4_interpolate}')
@@ -111,4 +112,5 @@ class VGenI2VScript(scripts_manager.Script):
shared.sd_model = orig_pipeline
if video_type != 'None' and processed is not None:
video.save_video(p, filename=None, images=processed.images, video_type=video_type, duration=duration, loop=gif_loop, pad=mp4_pad, interpolate=mp4_interpolate)
p.video_saved = True
return processed
+2
View File
@@ -69,6 +69,7 @@ class SVDScript(scripts_manager.Script):
return frames
def run(self, p: processing.StableDiffusionProcessing, model, num_frames, override_resolution, min_guidance_scale, max_guidance_scale, decode_chunk_size, motion_bucket_id, noise_aug_strength, video_type, duration, gif_loop, mp4_pad, mp4_interpolate): # pylint: disable=arguments-differ, unused-argument
p.video_interpolate = mp4_interpolate
image = getattr(p, 'init_images', None)
if image is None or len(image) == 0:
log.error('SVD: no init_images')
@@ -122,4 +123,5 @@ class SVDScript(scripts_manager.Script):
processed = processing.process_images(p)
if video_type != 'None':
video.save_video(p, filename=None, images=processed.images, video_type=video_type, duration=duration, loop=gif_loop, pad=mp4_pad, interpolate=mp4_interpolate)
p.video_saved = True
return processed
+2
View File
@@ -54,6 +54,7 @@ class ModelScopeScript(scripts_manager.Script):
def run(self, p: processing.StableDiffusionProcessing, model_name, use_default, num_frames, video_type, duration, gif_loop, mp4_pad, mp4_interpolate): # pylint: disable=arguments-differ, unused-argument
if model_name == 'None':
return None
p.video_interpolate = mp4_interpolate
model = [m for m in MODELS if m['name'] == model_name][0]
log.debug(f'Text2Video: model={model} defaults={use_default} frames={num_frames}, video={video_type} duration={duration} loop={gif_loop} pad={mp4_pad} interpolate={mp4_interpolate}')
@@ -90,4 +91,5 @@ class ModelScopeScript(scripts_manager.Script):
if video_type != 'None':
video.save_video(p, filename=None, images=processed.images, video_type=video_type, duration=duration, loop=gif_loop, pad=mp4_pad, interpolate=mp4_interpolate)
p.video_saved = True
return processed