hf-internal-testing

/

tiny-stable-diffusion-pipe

like 4

Follow

Hugging Face Internal Testing Organization 98

Hi all, not sure I am doing something wrong but this model doesn't seem to be working for me.

Check it out yourself with this colab

import os
import torch

from diffusers import StableDiffusionPipeline
from diffusers import DiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("hf-internal-testing/tiny-stable-diffusion-pipe", torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "a picture of a rabbit"
images = pipe(prompt, num_images_per_prompt=1).images

gives

RuntimeError: The size of tensor a (12545) must match the size of tensor b (226) at non-singleton dimension 1

https://colab.research.google.com/drive/1AE5uA68myGvyzEi-vm4w806EN9rdjzq4?usp=sharing