stable-diffusion AI 绘画

最新推荐文章于 2024-07-19 21:36:04 发布

code-thy

最新推荐文章于 2024-07-19 21:36:04 发布

阅读量188

点赞数

文章标签： stable diffusion 人工智能

本文链接：https://blog.csdn.net/weixin_57349625/article/details/131195791

版权

git clone https://github.com/CompVis/stable-diffusion.git
进入stable-diffusion目录
在这里注册一个账号: Hugging Face – The AI community building the future. 并生成个token
安装CUDA NVIDIA 安装 CUDA_nvidia cuda_长满头发的程序猿的博客-CSDN博客
pip install torch -f https://download.pytorch.org/whl/torch_stable.html
pip install -e git+ https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
conda install torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia 选择合适的pytorch-cuda版本
pip install transformers==4.25.1 diffusers invisible-watermark
pip install -e .
huggingface-cli login 输入生成的token

from torch import autocast
from diffusers import StableDiffusionPipeline
import torch

if __name__ == '__main__':
    # https://github.com/CompVis/stable-diffusion/issues/69#issuecomment-1260722801
    torch.backends.cudnn.benchmark = True
    torch.backends.cudnn.enabled = True

    pipe = StableDiffusionPipeline.from_pretrained(
        "CompVis/stable-diffusion-v1-4",
        torch_dtype=torch.float16,
        use_auth_token=True,
        safety_checker=None,
        requires_safety_checker=False
    ).to("cuda")


    prompt = "a photo of an astronaut riding a horse on mars"
    with autocast("cuda"):
        result = pipe(prompt, height=256, width=256)

    image = result[0][0]

    image.save("test.png")