AUTOMATIC1111/stable-diffusion-webui step4:模型库下载,下载的2.0版本,2.0版本模型文件共两个: 从https://huggingface.co/stabilityai/stable-diffusion
这个模型需要生成的图片在768768的时候效果比较好,如果是生成512512的,图片有点拉垮; # 768-v-ema.ckpt https://huggingface.co/stabilityai/stable-diffusion
优于篇幅太长,我们就不贴完整代码了,请在最后的完整代码中查看 模型加载与配置 我们将使用stable-diffusion-2 pretrained_model_name_or_path = "stabilityai /stable-diffusion-2" 然后就是与DreamBooth微调相关的配置参数包括加载模型体系结构的各种组件:文本编码器、变分自动编码器(VAE)和U-Net。
torch.float16, use_safetensors=True, variant="fp16") pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion
diffusion提供了一系列的模型: stable-diffusion-v1-4 擅长绘制风景类画,整体偏欧美风,具有划时代意义 stable-diffusion-v1-5 同上,但生成的作品更具艺术性 stable-diffusion
diffusion提供了一系列的模型: stable-diffusion-v1-4 擅长绘制风景类画,整体偏欧美风,具有划时代意义 stable-diffusion-v1-5 同上,但生成的作品更具艺术性 stable-diffusion
import os # Stable Diffusion API配置 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion
StableDiffusionPipeline # 加载Stable Diffusion模型 pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion
另外SD 2.0除了512x512版本的模型,还包括768x768版本的模型(https://huggingface.co/stabilityai/stable-diffusion-2),所谓的768x768