Skip to content

Commit 3e57c2e

Browse files
committed
feat(ltx2): add sharding section to config
1 parent c4d072b commit 3e57c2e

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

src/maxdiffusion/configs/ltx2_video.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,11 @@ logical_axis_rules: [
6464
]
6565
data_sharding: ['data', 'fsdp', 'context', 'tensor']
6666

67+
sharding:
68+
transformer: 'default'
69+
vae: 'default'
70+
text_encoder: 'default'
71+
6772
dcn_data_parallelism: 1 # recommended DCN axis to be auto-sharded
6873
dcn_fsdp_parallelism: -1
6974

0 commit comments

Comments
 (0)