rajabi2001

sega

rajabi2001

Official implementation of "SEGA: Spectral-Energy Guided Attention for Resolution Extrapolation in Diffusion Transformers"

AI 简介

SEGA 是一种无需训练、不修改模型权重或结构的注意力机制增强方法,专用于扩散Transformer(如FLUX和Qwen-Image)的高分辨率图像生成。其核心是基于潜空间频谱能量动态重标定RoPE位置编码各分量的注意力权重,在每步去噪中自适应提升高频细节重建能力。该方法支持4096×4096及以上超分辨率合成,并提供多GPU显存优化推理方案。适用于需在保持原模型前提下扩展生成分辨率的科研与工业图像生成场景。

Python
67
Stars
6
Forks
2
Watchers
2
Issues

Star 增长

今日0
近 7 天0
近 30 天0
综合评分2.54
默认分支main