SD3模型的部署（本地部署）
文章目录
- 模型权重的下载
- 需要注意的地方
- 推理代码
- 生成的效果图
模型的结构图
模型权重的下载
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 StableDiffusion3Pipeline
pipe = 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
生成的效果图
提示词是：一只红色的可爱，活泼小熊，生成的图像如下：
提示词是：晨光熹微，湖面波光粼粼，鸟语花香，心旷神怡。
