一种扩散模型融合方法、装置及电子设备

本申请公开了一种扩散模型融合方法、装置及电子设备。涉及人工智能领域，该方法包括：获取文本信息，其中，文本信息中包括待生成目标图像的描述信息；确定与文本信息相匹配的目标风格提示向量，其中，目标风格提示向量用于指示目标学生模型输出目标图像风格的图像，目标学生模型是在分布式训练框架下基于多种不同图像风格的教师模型进行知识蒸馏得到的学生模型；将文本信息和目标风格提示向量输入至目标学生模型，由目标学生模型执行扩散生成过程，输出目标图像。通过本申请，解决了相关技术中现有的模型融合方法得到的融合模型在生成不同风格的图像时，生成的图像质量较低的问题。