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

support value controller training

Browse Source
This commit is contained in:
Artiprocher
2025-07-21 19:16:30 +08:00
parent 22705a44b4
commit e3c5d2540b
7 changed files with 79 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/flux/model_inference/FLUX.1-dev-AttriCtrl.py
View File

@@ -15,5 +15,5 @@ pipe = FluxImagePipeline.from_pretrained(
)
for i in [0.1, 0.3, 0.5, 0.7, 0.9]:
image = pipe(prompt="A woman.", seed=602, value_controller_inputs=[i], rand_device="cuda")
image = pipe(prompt="a cat on the beach", seed=2, value_controller_inputs=[i])
image.save(f"value_control_{i}.jpg")