Stable Diffusion中文网是基于LAION-5B数据集进行训练的,该数据集来源于从网络上抓取的Common Crawl数据。LAION-5B是一个公开的数据集,由德国非营利组织LAION创建,并得到了StabilityAI的资助。训练过程中,模型首先在LAION-5B的一个大子集上进行训练,然后在“LAION-Aesthetics v2 5+”上进行最后几轮训练。这个子集包含了6亿张带标题的图片,人工智能预测人类对这些图片的喜欢程度至少为5/10。为了保证训练数据的质量,低分辨率的图像和带有水印的图像被排除在最终的子集之外。
模型的训练过程使用了亚马逊云计算服务上的256个NVIDIA A100 GPU,总共花费了15万个GPU小时,成本为60万美元。
为了纠正模型初始训练的局限性,Stable Diffusion中文网提供了终端用户微调训练的功能。用户可以选择对模型权重进行微调,以生成更符合特定使用情境的输出。以下是三种用户可以使用的微调方法:
通过这些微调方法,终端用户可以根据自己的需求和偏好对Stable Diffusion模型进行个性化的调整。