nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[hotfix] update benchmark for ddp (#23)

Browse source
This commit is contained in:
Hongxin Liu 2024-03-04 14:31:41 +08:00 committed by GitHub
parent 9648d53d4d
commit a05e2a3d68
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
benchmark.py
View file

@ -56,7 +56,7 @@ def main(args):
video_compressor = create_video_compressor(args.compressor)
model_kwargs = {
"in_channels": video_compressor.out_channels,
"seq_parallel_group": plugin.sp_group,
"seq_parallel_group": getattr(plugin, "sp_group", None),
"seq_parallel_mode": args.sp_mode,
"seq_parallel_overlap": args.sp_overlap,
}

4
open_sora/modeling/dit/__init__.py
View file

@ -1,3 +1,3 @@
from .dit import DiT, DiT_models
from .dit import SUPPORTED_SEQ_PARALLEL_MODES, DiT, DiT_models
__all__ = ["DiT_models", "DiT"]
__all__ = ["DiT_models", "DiT", "SUPPORTED_SEQ_PARALLEL_MODES"]