Update scripts/gradio/i2v_test_application.py

scripts/gradio/i2v_test_application.py

@@ -41,7 +41,7 @@ class Image2Video():
     def __init__(self,result_dir='./tmp/',gpu_num=1,resolution='256_256') -> None:
         self.resolution = (int(resolution.split('_')[0]), int(resolution.split('_')[1])) #hw
         self.download_model()
-        
+        print("after download model")
         self.result_dir = result_dir
         if not os.path.exists(self.result_dir):
             os.mkdir(self.result_dir)
@@ -59,8 +59,9 @@ class Image2Video():
         cn_config = OmegaConf.load(cn_config_file)
         cn_model_config = cn_config.pop("control_stage_config", OmegaConf.create())
 
- 
+        print("before init") 
         model_list = []
+
         for gpu_id in range(gpu_num):
             model = instantiate_from_config(model_config)
             cn_model = instantiate_from_config(cn_model_config)
@@ -78,6 +79,7 @@ class Image2Video():
             model_list.append(model)
         self.model_list = model_list
         self.save_fps = 8
+        print("init done.")
 
     @spaces.GPU(duration=100)
     def get_image(self, image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, image2=None, frame_guides=None,control_scale=0.6):