Reduce chunk size to be able to run on A10G

app_canny.py

@@ -47,7 +47,7 @@ def create_demo(model: Model):
                     watermark = gr.Radio(["Picsart AI Research", "Text2Video-Zero",
                                          "None"], label="Watermark", value='Picsart AI Research')
                     chunk_size = gr.Slider(
-                        label="Chunk size", minimum=2, maximum=16, value=8, step=1, visible=not on_huggingspace,
+                        label="Chunk size", minimum=2, maximum=16, value=2, step=1, visible=not on_huggingspace,
                         info="Number of frames processed at once. Reduce for lower memory usage.")
                     merging_ratio = gr.Slider(
                         label="Merging ratio", minimum=0.0, maximum=0.9, step=0.1, value=0.0, visible=not on_huggingspace,

app_canny_db.py

@@ -51,7 +51,7 @@ def create_demo(model: Model):
                     watermark = gr.Radio(["Picsart AI Research", "Text2Video-Zero",
                                          "None"], label="Watermark", value='Picsart AI Research')
                     chunk_size = gr.Slider(
-                        label="Chunk size", minimum=2, maximum=16, value=8, step=1, visible=not on_huggingspace,
+                        label="Chunk size", minimum=2, maximum=16, value=2, step=1, visible=not on_huggingspace,
                         info="Number of frames processed at once. Reduce for lower memory usage.")
                     merging_ratio = gr.Slider(
                         label="Merging ratio", minimum=0.0, maximum=0.9, step=0.1, value=0.0, visible=not on_huggingspace,

app_depth.py

@@ -60,7 +60,7 @@ def create_demo(model: Model):
                     watermark = gr.Radio(["Picsart AI Research", "Text2Video-Zero",
                                          "None"], label="Watermark", value='Picsart AI Research')
                     chunk_size = gr.Slider(
-                        label="Chunk size", minimum=2, maximum=16, value=8, step=1, visible=not on_huggingspace,
+                        label="Chunk size", minimum=2, maximum=16, value=2, step=1, visible=not on_huggingspace,
                         info="Number of frames processed at once. Reduce for lower memory usage.")
                     merging_ratio = gr.Slider(
                         label="Merging ratio", minimum=0.0, maximum=0.9, step=0.1, value=0.0, visible=not on_huggingspace,

app_pix2pix_video.py

@@ -74,7 +74,7 @@ def create_demo(model: Model):
                                         value=-1,
                                         step=1)
                     chunk_size = gr.Slider(
-                        label="Chunk size", minimum=2, maximum=16, value=8, step=1, visible=not on_huggingspace,
+                        label="Chunk size", minimum=2, maximum=16, value=2, step=1, visible=not on_huggingspace,
                         info="Number of frames processed at once. Reduce for lower memory usage.")
                     merging_ratio = gr.Slider(
                         label="Merging ratio", minimum=0.0, maximum=0.9, step=0.1, value=0.0, visible=not on_huggingspace,

app_pose.py

@@ -35,7 +35,7 @@ def create_demo(model: Model):
                     watermark = gr.Radio(["Picsart AI Research", "Text2Video-Zero",
                                          "None"], label="Watermark", value='Picsart AI Research')
                     chunk_size = gr.Slider(
-                        label="Chunk size", minimum=2, maximum=16, value=8, step=1, visible=not on_huggingspace,
+                        label="Chunk size", minimum=2, maximum=16, value=2, step=1, visible=not on_huggingspace,
                         info="Number of frames processed at once. Reduce for lower memory usage.")
                     merging_ratio = gr.Slider(
                         label="Merging ratio", minimum=0.0, maximum=0.9, step=0.1, value=0.0, visible=not on_huggingspace,

app_text_to_video.py

@@ -80,7 +80,7 @@ def create_demo(model: Model):
                                    info="Perform DDPM steps from t0 to t1. The larger the gap between t0 and t1, the more variance between the frames. Ensure t0 < t1",
                                    maximum=48, value=47, step=1)
                     chunk_size = gr.Slider(
-                        label="Chunk size", minimum=2, maximum=16, value=8, step=1, visible=not on_huggingspace,
+                        label="Chunk size", minimum=2, maximum=16, value=2, step=1, visible=not on_huggingspace,
                         info="Number of frames processed at once. Reduce for lower memory usage."
                     )
                     merging_ratio = gr.Slider(