Skip to content

Commit cf40333

Browse files
committed
Fix linting
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
1 parent cf44aa0 commit cf40333

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

examples/automodel/pretrain/wan2_1_t2v_flow.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,12 @@ dist_env:
1111

1212
model:
1313
pretrained_model_name_or_path: Wan-AI/Wan2.1-T2V-1.3B-Diffusers
14-
mode: pretrain
14+
mode: pretrain
1515
pipeline_spec:
16-
transformer_cls: "WanTransformer3DModel"
17-
subfolder: "transformer"
16+
transformer_cls: "WanTransformer3DModel"
17+
subfolder: "transformer"
1818
enable_gradient_checkpointing: false
19-
load_full_pipeline: false
19+
load_full_pipeline: false
2020

2121
step_scheduler:
2222
global_batch_size: 8

0 commit comments

Comments
 (0)