update empty_cache

app.py

@@ -41,7 +41,8 @@ def videocrafter_demo(result_dir='./tmp/'):
                             fps = gr.Slider(minimum=4, maximum=32, step=1, label='fps', value=16, elem_id="fps")
                         send_btn = gr.Button("Send")
                     with gr.Tab(label='result'):
-                        output_video_1 =  gr.Video().style(width=320)
+                        with gr.Row():
+                            output_video_1 =  gr.Video().style(width=512)
                 gr.Examples(examples=t2v_examples,
                             inputs=[input_text,steps,cfg_scale,eta],
                             outputs=[output_video_1],
@@ -71,7 +72,7 @@ def videocrafter_demo(result_dir='./tmp/'):
                         i2v_end_btn = gr.Button("Send")
                     with gr.Tab(label='Result'):
                         with gr.Row():
-                            i2v_output_video = gr.Video(label="Generated Video").style(width=320)
+                            i2v_output_video = gr.Video(label="Generated Video").style(width=512)
 
                 gr.Examples(examples=i2v_examples,
                             inputs=[i2v_input_image, i2v_input_text, i2v_steps, i2v_cfg_scale, i2v_eta, i2v_fps],

i2v_test.py

@@ -30,6 +30,7 @@ class Image2Video():
         self.save_fps = 8
 
     def get_image(self, image, prompt, steps=50, cfg_scale=12.0, eta=1.0, fps=16):
+        torch.cuda.empty_cache()
         gpu_id=0
         if steps > 60:
             steps = 60 

t2v_test.py

@@ -28,6 +28,7 @@ class Text2Video():
         self.save_fps = 8
 
     def get_prompt(self, prompt, steps=50, cfg_scale=12.0, eta=1.0, fps=16):
+        torch.cuda.empty_cache()
         gpu_id=0
         if steps > 60:
             steps = 60