huggingface / huggingface/diffusers

[New feature] A Noise Injection Method for Flux

Đang mở
#10,071 10 bình luận 1 reaction 0 người được giao Xem trên GitHub
consider-for-modular-diffusers wip
Ngôn ngữ chính
Python
Star
34.5k
Fork
7.3k
Merge trung bình
3 ngày 3 giờ
Pull request đã merge (30 ngày)
91

Mô tả

### New Feature for FluxPipeline
Paper: [Link](https://openreview.net/pdf?id=KzokzKV4JK)
Code : [Link](https://anonymous.4open.science/r/SSNI-F746/README.md)
workflow : [Link](https://www.dropbox.com/scl/fi/hhitjx6lqpqpv8xjx9ikq/Flux-Noise-Injection.json?rlkey=45xnu45j1i5owiwhc7z1hppn1&e=1&dl=0)
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](https://github.com/user-attachments/assets/e87d3da6-6ff5-46c1-b0ff-20c00ad4e541)
![spider_sh](https://github.com/user-attachments/assets/e0f7a469-ae99-4095-9c9d-8cc88ed4691d)

@sayakpaul @yiyixuxu

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Hướng nghiên cứu

Bắt đầu với FluxPipeline và đọc paper được liên kết, implementation tham chiếu cũng như workflow để hiểu phương pháp SSNI được đề xuất. Công việc được xem là hoàn tất khi có một tính năng tích hợp để chèn nhiễu cho Flux, kèm theo việc validation dựa trên hành vi và các kết quả được mô tả trong issue.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Đánh giá

Công nghệ
python, pytorch
Lĩnh vực
machine-learning
Loại issue
Tính năng
Độ khó
5/5
Thời gian dự kiến
Hơn một tuần
Mức độ hoạt động
Đình trệ
Độ rõ ràng
Cần làm rõ
Mức phù hợp với người mới
25/100

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.