Update inference/flovd_demo.py

inference/flovd_demo.py

@@ -342,13 +342,13 @@ def generate_video(
         target_modules=["to_q", "to_k", "to_v", "to_out.0", "norm1.linear", "norm2.linear", "ff.net.2"],
     )
     
-    print(f'Constructing pipeline')
+    print(f'Constructing pipeline', flush=True)
     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')
+    print(f'Done loading pipeline', flush=True)
     
     """
         # Prepare inputs