Update inference/flovd_demo.py

inference/flovd_demo.py

@@ -344,13 +344,20 @@ def generate_video(
     
     print(f'Constructing pipeline')
     pipe_omsm = load_cogvideox_flovd_OMSM_lora_pipeline(omsm_path, backbone_path="THUDM/CogVideoX-5b-I2V", transformer_lora_config=transformer_lora_config, device=local_rank, dtype=dtype)   
+    print("done with omsm", flush=True)
     pipe_fvsm = load_cogvideox_flovd_FVSM_controlnet_pipeline(fvsm_path, backbone_path="THUDM/CogVideoX-5b-I2V", device=local_rank, dtype=dtype)   
+    print("done with fvsm", flush=True)
+
     print(f'Done loading pipeline')
     
     """
         # Prepare inputs
     """
+    print(f'loading image', flush=True)
+    
     image = load_image(image=image_path)
+    print(f'done loading image', flush=True)
+    
     
     assert pose_type in ['re10k', 'manual'], "Choose other pose_type between ['re10k', 'manual']"
     if pose_type == 're10k':