We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent cf44aa0 commit cf40333Copy full SHA for cf40333
examples/automodel/pretrain/wan2_1_t2v_flow.yaml
@@ -11,12 +11,12 @@ dist_env:
11
12
model:
13
pretrained_model_name_or_path: Wan-AI/Wan2.1-T2V-1.3B-Diffusers
14
- mode: pretrain
+ mode: pretrain
15
pipeline_spec:
16
- transformer_cls: "WanTransformer3DModel"
17
- subfolder: "transformer"
+ transformer_cls: "WanTransformer3DModel"
+ subfolder: "transformer"
18
enable_gradient_checkpointing: false
19
- load_full_pipeline: false
+ load_full_pipeline: false
20
21
step_scheduler:
22
global_batch_size: 8
0 commit comments