textboost

一张图微调文本编码器，让 Stable Diffusion 学会你的专属主体

TextBoost 是一个面向文本到图像模型的个性化微调工具，核心解决单张参考图即可实现主体定制化生成的问题。传统方法如 DreamBooth 需要多张图像并微调整个 UNet，计算成本高且易过拟合；TextBoost 仅微调文本编码器，通过优化文本嵌入空间来绑定特定主体概念，无需修改扩散模型主体结构。项目基于 PyTorch 和 Stable Diffusion，提供 Jupyter Notebook 形式的完整实验流程，支持用户用一张照片训练模型，在推理时通过自然语言描述场景，生成包含该主体的新图像。其核心能力包括：单样本学习、低资源训练（仅更新文本编码器参数）、与现有 ControlNet 等插件兼容，适合快速原型验证和学术研究。