rajabi2001

sega

rajabi2001

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

AI 简介

SEGA项目实现了“基于频谱能量引导的注意力机制以提高扩散变换器中的分辨率外推”。其核心功能是通过动态调整RoPE组件的注意力比例来根据潜在空间的频率内容在每个去噪步骤中优化高分辨率图像合成,无需重新训练或修改架构。该项目使用Python编写,并提供了FLUX和Qwen-Image两种实现方式。特别适用于需要生成超高清图像但受限于单GPU内存容量的场景,支持多GPU并行处理以扩展计算能力。

Python
最后更新 2026年5月28日活跃
66
Stars
5
Forks
2
Watchers
2
Issues

Star 增长

今日0
近 7 天+7
近 30 天+9
综合评分43.73
默认分支main