diff --git a/nodes.py b/nodes.py index 3f88d93..debc2d1 100644 --- a/nodes.py +++ b/nodes.py @@ -84,7 +84,6 @@ class DynamiCrafterI2V: "optional": { "image2": ("IMAGE",), "mask": ("MASK",), - "looping": ("BOOLEAN", {"default": False}), } } @@ -93,7 +92,7 @@ class DynamiCrafterI2V: FUNCTION = "process" CATEGORY = "DynamiCrafterWrapper" - def process(self, model, image, prompt, cfg, steps, eta, seed, fs, keep_model_loaded, frames, mask=None, image2=None, looping=False): + def process(self, model, image, prompt, cfg, steps, eta, seed, fs, keep_model_loaded, frames, mask=None, image2=None): device = mm.get_torch_device() mm.unload_all_models() mm.soft_empty_cache() @@ -132,10 +131,6 @@ class DynamiCrafterI2V: img_tensor_repeat[:,:,-1:,:,:] = z2 else: img_tensor_repeat = repeat(z, 'b c t h w -> b c (repeat t) h w', repeat=frames) - if looping: - img_tensor_repeat = torch.zeros_like(img_tensor_repeat) - img_tensor_repeat[:,:,:1,:,:] = z - img_tensor_repeat[:,:,-1:,:,:] = z self.model.first_stage_model.to('cpu')