git clone https://github.com/CompVis/stable-diffusion.git
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
输入生成的tokenfrom 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")
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。