theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-19 06:48:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

update doc

Browse Source
This commit is contained in:
Artiprocher
2025-11-10 10:05:19 +08:00
parent ceb473efc0
commit eacec13309
13 changed files with 489 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/diffusion/runner.py
View File

@@ -12,7 +12,7 @@ def launch_training_task(
model_logger: ModelLogger,
learning_rate: float = 1e-5,
weight_decay: float = 1e-2,
num_workers: int = 8,
num_workers: int = 1,
save_steps: int = None,
num_epochs: int = 1,
args = None,

4
diffsynth/diffusion/training_module.py
View File

@@ -115,8 +115,8 @@ class DiffusionTrainingModule(torch.nn.Module):
def switch_pipe_to_training_mode(
self,
pipe,
trainable_models,
lora_base_model, lora_target_modules, lora_rank, lora_checkpoint=None,
trainable_models=None,
lora_base_model=None, lora_target_modules="", lora_rank=32, lora_checkpoint=None,
preset_lora_path=None, preset_lora_model=None,
):
# Scheduler