文本生成图片 学习笔记

2023-12-13 21:56:09

目录

diffusion-v1-5

修改参数sample_size:


diffusion-v1-5

安装:

pip install diffusers

模型下载地址:

深入浅出完整解析Stable Diffusion(SD)核心基础知识 - 知乎

from diffusers import StableDiffusionPipeline
import torch

model_id = "./stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "a

文章来源:https://blog.csdn.net/jacke121/article/details/134981149
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。