网站企业推广方案电工培训学校

模型的结构图

模型权重的下载

SD3：huggingface的权重
我们需要把huggingfaceface下的这些文件都下载到一个文件加下，然后在后面的pipe = StableDiffusion3Pipeline.from_pretrained(“stabilityai/stable-diffusion-3-medium-diffusers”, torch_dtype=torch.float16)这里传入这个文件夹的路径

需要注意我们的第三个文本编码器器我们需要下的是这些东西

这里是我打包好的SD3的权重（少了第三个编码器的一个safesentor）

需要注意的地方

需要安装的一些包

protobuf
sentencepiece

推理代码

import torch
from diffusers import StableDiffusion3Pipelinepipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
pipe = pipe.to("cuda")image = pipe("A cat holding a sign that says hello world",negative_prompt="",num_inference_steps=28,guidance_scale=7.0,
).images[0]
image

生成的效果图

提示词是：一只红色的可爱，活泼小熊，生成的图像如下：

提示词是：晨光熹微，湖面波光粼粼，鸟语花香，心旷神怡。