stable-diffusion icon indicating copy to clipboard operation
stable-diffusion copied to clipboard

Published 20 hours ago • verified publisher icon runwayml

RuntimeError: expected scalar type Half but found Float

Open fallbernana123456 opened this issue 1 year ago • 1 comments

from diffusers import StableDiffusionPipeline

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

prompt = "a photo of an astronaut riding a horse on mars" image = pipe(prompt).images[0]

transformers/models/clip/modeling_clip.py:257 in forward

fallbernana123456 avatar Apr 20 '23 05:04 fallbernana123456

add this one with torch.autocast("cuda"):

weiguangzhao avatar Apr 28 '23 11:04 weiguangzhao