update log time

i2v_test.py

@@ -1,4 +1,5 @@
 import os
+import time
 from omegaconf import OmegaConf
 
 import torch
@@ -31,6 +32,8 @@ class Image2Video():
 
     def get_image(self, image, prompt, steps=50, cfg_scale=12.0, eta=1.0, fps=16):
         torch.cuda.empty_cache()
+        print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
+        start = time.time()
         gpu_id=0
         if steps > 60:
             steps = 60 
@@ -61,6 +64,7 @@ class Image2Video():
         prompt_str=prompt_str[:30]
 
         save_videos(batch_samples, self.result_dir, filenames=[prompt_str], fps=self.save_fps)
+        print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
         return os.path.join(self.result_dir, f"{prompt_str}.mp4")
     
     def download_model(self):

t2v_test.py

@@ -1,4 +1,5 @@
 import os
+import time
 from omegaconf import OmegaConf
 import torch
 from scripts.evaluation.funcs import load_model_checkpoint, save_videos, batch_ddim_sampling
@@ -29,6 +30,8 @@ class Text2Video():
 
     def get_prompt(self, prompt, steps=50, cfg_scale=12.0, eta=1.0, fps=16):
         torch.cuda.empty_cache()
+        print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
+        start = time.time()
         gpu_id=0
         if steps > 60:
             steps = 60 
@@ -52,6 +55,7 @@ class Text2Video():
         prompt_str=prompt_str[:30]
 
         save_videos(batch_samples, self.result_dir, filenames=[prompt_str], fps=self.save_fps)
+        print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
         return os.path.join(self.result_dir, f"{prompt_str}.mp4")
     
     def download_model(self):