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

Merge pull request #263 from modelscope/super-alignment

Browse Source 
support mask blur
This commit is contained in:
Zhongjie Duan
2024-11-11 19:24:30 +08:00
committed by GitHub
parent a7050a185b 8d1d1536d3
commit 75bba5b8e5
3 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/train/flux/train_flux_lora.py
View File

@@ -22,7 +22,9 @@ class LightningModel(LightningModelForT2ILoRA):
model_manager.load_models(pretrained_weights[1:])
model_manager.load_model(pretrained_weights[0], torch_dtype=quantize)
if preset_lora_path is not None:
model_manager.load_lora(preset_lora_path)
preset_lora_path = preset_lora_path.split(",")
for path in preset_lora_path:
model_manager.load_lora(path)
self.pipe = FluxImagePipeline.from_model_manager(model_manager)