gigagan-pytorch

收录理由

Adobe 推出的 GigaGAN 把基于 GAN 的图像合成推到了新的高度。这个仓库用 PyTorch 实现得清晰易读，而且带有可直接运行的训练流程。对于不想转向扩散模型、又想训练现代 GAN 的研究者或工程师来说，这里有无条件生成器、1k 到 4k 的上采样器，还能通过 CLIP 嵌入做可选的文本条件控制。代码里加入了 skip-layer excitation 来加速收敛，还有 reconstruction auxiliary loss 提升稳定性，并且借助 Hugging Face Accelerate 支持多 GPU 训练。用 pip 安装后，指向一个图片文件夹，就能训练 256x256 的生成器。不过它更偏研究工具，需要自己跑训练，而不是下载现成的预训练模型。