chunk size set to 2

model.py

@@ -83,7 +83,7 @@ class Model:
                          generator=self.generator,
                          **kwargs)
 
-    def inference(self, split_to_chunks=False, chunk_size=8, **kwargs):
+    def inference(self, split_to_chunks=False, chunk_size=2, **kwargs):
         if not hasattr(self, "pipe") or self.pipe is None:
             return
         '''
@@ -135,7 +135,7 @@ class Model:
     def process_controlnet_canny(self,
                                  video_path,
                                  prompt,
-                                 chunk_size=8,
+                                 chunk_size=2,
                                  watermark='Picsart AI Research',
                                  merging_ratio=0.0,
                                  num_inference_steps=20,
@@ -200,7 +200,7 @@ class Model:
     def process_controlnet_depth(self,
                                  video_path,
                                  prompt,
-                                 chunk_size=8,
+                                 chunk_size=2,
                                  watermark='Picsart AI Research',
                                  merging_ratio=0.0,
                                  num_inference_steps=20,
@@ -263,7 +263,7 @@ class Model:
     def process_controlnet_pose(self,
                                 video_path,
                                 prompt,
-                                chunk_size=8,
+                                chunk_size=2,
                                 watermark='Picsart AI Research',
                                 merging_ratio=0.0,
                                 num_inference_steps=20,
@@ -326,7 +326,7 @@ class Model:
                                     db_path,
                                     video_path,
                                     prompt,
-                                    chunk_size=8,
+                                    chunk_size=2,
                                     watermark='Picsart AI Research',
                                     merging_ratio=0.0,
                                     num_inference_steps=20,
@@ -397,7 +397,7 @@ class Model:
                         start_t=0,
                         end_t=-1,
                         out_fps=-1,
-                        chunk_size=8,
+                        chunk_size=2,
                         watermark='Picsart AI Research',
                         merging_ratio=0.0,
                         use_cf_attn=True,
@@ -434,7 +434,7 @@ class Model:
                            t0=44,
                            t1=47,
                            n_prompt="",
-                           chunk_size=8,
+                           chunk_size=2,
                            video_length=8,
                            watermark='Picsart AI Research',
                            merging_ratio=0.0,