Remove looping option as it's reduntant

Results are exact same when simply using same image in both image inputs.
This commit is contained in:
kijai
2024-03-17 00:02:01 +02:00
parent 730ab19139
commit 35f6837dab
+1 -6
View File
@@ -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')