From 923f55e2134f55c0fd495756cff2f7e7a3840941 Mon Sep 17 00:00:00 2001 From: Jade Zheng Date: Wed, 4 Dec 2024 18:12:32 +0800 Subject: [PATCH] fix: invalid call to is_context_parallel_initialized The function is_context_parallel_initialized was not actually being called in the original implementation. This commit corrects the issue by ensuring the function is properly invoked. --- sat/vae_modules/autoencoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sat/vae_modules/autoencoder.py b/sat/vae_modules/autoencoder.py index 0adad85..e90a0d0 100644 --- a/sat/vae_modules/autoencoder.py +++ b/sat/vae_modules/autoencoder.py @@ -578,7 +578,7 @@ class VideoAutoencoderInferenceWrapper(VideoAutoencodingEngine): if self.cp_size <= 1: use_cp = False if self.cp_size > 0 and use_cp and not input_cp: - if not is_context_parallel_initialized: + if not is_context_parallel_initialized(): initialize_context_parallel(self.cp_size) global_src_rank = get_context_parallel_group_rank() * self.cp_size @@ -609,7 +609,7 @@ class VideoAutoencoderInferenceWrapper(VideoAutoencodingEngine): if self.cp_size <= 1: use_cp = False if self.cp_size > 0 and use_cp and not input_cp: - if not is_context_parallel_initialized: + if not is_context_parallel_initialized(): initialize_context_parallel(self.cp_size) global_src_rank = get_context_parallel_group_rank() * self.cp_size