diff --git a/CHANGELOG.md b/CHANGELOG.md index 7640a9655..0ab6a5880 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -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 diff --git a/html/reference.json b/html/reference.json index b21dd95f7..fc070b320 100644 --- a/html/reference.json +++ b/html/reference.json @@ -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", diff --git a/pipelines/flux/flux_nunchaku.py b/pipelines/flux/flux_nunchaku.py index 1aba177aa..49fd77b3d 100644 --- a/pipelines/flux/flux_nunchaku.py +++ b/pipelines/flux/flux_nunchaku.py @@ -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"