网站快照工具,google应用商店,5118网站的功能,网站建设要学习什么文章目录 模型权重的下载需要注意的地方推理代码生成的效果图 模型的结构图 模型权重的下载
SD3#xff1a;huggingface的权重 我们需要把huggingfaceface下的这些文件都下载到一个文件加下#xff0c;然后在后面的pipe StableDiffusion3Pipeline.from_pretrained(“stabil… 文章目录 模型权重的下载需要注意的地方推理代码生成的效果图 模型的结构图 模型权重的下载
SD3huggingface的权重 我们需要把huggingfaceface下的这些文件都下载到一个文件加下然后在后面的pipe StableDiffusion3Pipeline.from_pretrained(“stabilityai/stable-diffusion-3-medium-diffusers”, torch_dtypetorch.float16)这里传入这个文件夹的路径 需要注意我们的第三个文本编码器器我们需要下的是这些东西 这里是我打包好的SD3的权重少了第三个编码器的一个safesentor
需要注意的地方
需要安装的一些包
protobuf
sentencepiece推理代码
import torch
from diffusers import StableDiffusion3Pipelinepipe StableDiffusion3Pipeline.from_pretrained(stabilityai/stable-diffusion-3-medium-diffusers, torch_dtypetorch.float16)
pipe pipe.to(cuda)image pipe(A cat holding a sign that says hello world,negative_prompt,num_inference_steps28,guidance_scale7.0,
).images[0]
image
生成的效果图
提示词是一只红色的可爱活泼小熊生成的图像如下
提示词是晨光熹微湖面波光粼粼鸟语花香心旷神怡。
