theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-18 22:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

merge data process to training script

Browse Source
This commit is contained in:
Artiprocher
2025-09-04 15:18:56 +08:00
parent cb8de6be1b
commit 144365b07d
6 changed files with 35 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/wanvideo/model_training/train.py
View File

@@ -26,7 +26,7 @@ class WanTrainingModule(DiffusionTrainingModule):
# Training mode
self.switch_pipe_to_training_mode(
self, self.pipe, trainable_models,
self.pipe, trainable_models,
lora_base_model, lora_target_modules, lora_rank, lora_checkpoint=lora_checkpoint,
enable_fp8_training=False,
)