add flux.1-dev-srpo

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-09-18 13:36:29 -04:00
parent f971550bfb
commit 710983d098
3 changed files with 12 additions and 1 deletions
+2
View File
@@ -5,6 +5,8 @@
- **Models**
- [WAN 2.2 14B VACE](https://huggingface.co/alibaba-pai/Wan2.2-VACE-Fun-A14B)
available for *text-to-image* and *text-to-video* and *image-to-video* workflows
- [Tencent FLUX.1 Dev SRPO](https://huggingface.co/tencent/SRPO)
SRPO is trained by with specific technique: Directly Aligning the Full Diffusion Trajectory with Fine-Grained Human Preference
- **Offloading**
- improve offloading for pipelines with multiple stages such as *wan-2.2-14b*
- add timers to measure onload/offload times during generate
+7
View File
@@ -138,6 +138,13 @@
"skip": true,
"extras": "sampler: Default, cfg_scale: 4.5"
},
"Tencent FLUX.1 Dev SRPO": {
"path": "vladmandic/flux.1-dev-SRPO",
"preview": "vladmandic--flux.1-dev-SRPO.jpg",
"desc": "FLUX.1 Dev SRPO is Tencent trained with specific technique: Directly Aligning the Full Diffusion Trajectory with Fine-Grained Human Preference",
"skip": true,
"extras": "sampler: Default, cfg_scale: 4.5"
},
"Qwen-Image": {
"path": "Qwen/Qwen-Image",
+3 -1
View File
@@ -6,7 +6,9 @@ def load_flux_nunchaku(repo_id):
nunchaku_precision = nunchaku.utils.get_precision()
nunchaku_repo = None
transformer = None
if 'flux.1-dev' in repo_id.lower():
if 'srpo' in repo_id.lower():
pass
elif 'flux.1-dev' in repo_id.lower():
nunchaku_repo = f"nunchaku-tech/nunchaku-flux.1-dev/svdq-{nunchaku_precision}_r32-flux.1-dev.safetensors"
elif 'flux.1-schnell' in repo_id.lower():
nunchaku_repo = f"nunchaku-tech/nunchaku-flux.1-schnell/svdq-{nunchaku_precision}_r32-flux.1-schnell.safetensors"