Skip to content

[New feature] A Noise Injection Method for Flux #10071

Open
@SahilCarterr

Description

@SahilCarterr

New Feature for FluxPipeline

Paper: Link
Code : Link
workflow : Link
This paper introduces Sample-specific Score-aware Noise Injection (SSNI) to improve diffusion-based purification (DBP) methods. Unlike existing approaches that use a fixed noise level (t*) for all samples, SSNI adapts t* based on how noisy or clean each sample is. Using a pre-trained score network, SSNI estimates a sample's deviation from the clean data and adjusts the noise level accordingly.

This have stunning Results with Flux

spider_with_noise_injection
spider_sh

@sayakpaul @yiyixuxu

Metadata

Metadata

Assignees

No one assigned

    Labels

    consider-for-modular-diffusersThings to consider adding support for in Modular Diffusers (with the help of community)wip

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions