JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-12-02 18:52:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: invalid call to is_context_parallel_initialized

Browse Source 
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.
This commit is contained in:
Jade Zheng 2024-12-04 18:12:32 +08:00
parent 5aa6d3a9ee
commit 923f55e213
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sat/vae_modules/autoencoder.py
View File

@ -578,7 +578,7 @@ class VideoAutoencoderInferenceWrapper(VideoAutoencodingEngine):
if self.cp_size <= 1: if self.cp_size <= 1:
use_cp = False use_cp = False
if self.cp_size > 0 and use_cp and not input_cp: 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) initialize_context_parallel(self.cp_size)
global_src_rank = get_context_parallel_group_rank() * 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: if self.cp_size <= 1:
use_cp = False use_cp = False
if self.cp_size > 0 and use_cp and not input_cp: 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) initialize_context_parallel(self.cp_size)
global_src_rank = get_context_parallel_group_rank() * self.cp_size global_src_rank = get_context_parallel_group_rank() * self.cp_size