Update kandinsky/t2v_pipeline.py

kandinsky/t2v_pipeline.py

@@ -113,7 +113,7 @@ class Kandinsky4T2VPipeline:
             seed: int = None,
             return_frames: bool = False
     ):
-        num_steps = 4
+        num_steps = 8
 
         # SEED 
         if seed is None:
@@ -135,7 +135,7 @@ class Kandinsky4T2VPipeline:
         if (height, width) not in self.RESOLUTIONS[self.resolution]:
             raise ValueError(f"Wrong height, width pair. Available (height, width) are: {self.RESOLUTIONS[self.resolution]}")
 
-        if num_steps != 4:
+        if num_steps != 8:
             raise NotImplementedError(f"In the distilled version number of steps have to be strictly equal to 4")
 
         # PREPARATION