Bingsu

/

my-korean-stable-diffusion-v1-5

like 23

import torch
from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained("Bingsu/my-korean-stable-diffusion-v1-5")

pipe.to("cuda")
prompt = "cat"
seed = 46399
generator = torch.Generator("cuda").manual_seed(seed)
pipe(prompt, num_inference_steps=25, generator=generator).images[0]

모델 카드에 있는 코드 그대로 했는데 실행이 안됩니다. 다른 환경에서도 해봐도 똑같은 에러가 발생합니다.