RuntimeError: mat1 and mat2 shapes cannot be multiplied (154x512 and 768x320)

#11
by CNT73 - opened

Hi Guys i try to run this model (Realistic_Vision) locally but when i try to generate image i have this error : RuntimeError: mat1 and mat2 shapes cannot be multiplied (154x512 and 768x320). Any idea what I;m doing in wrong way. This is my code :

from diffusers import StableDiffusionPipeline
import torch
import uuid
import os

model_path = r"D:\Stabble\real6" #This is folder where i placed files downloaded from https://huggingface.co/SG161222/Realistic_Vision_V6.0_B1_noVAE/tree/main

pipe = StableDiffusionPipeline.from_pretrained(model_path)
pipe.to("cpu")

Generate the image with the specified size

image = pipe("A whole body image of flying dragon", num_inference_steps=50, height=768, width=768).images[0]

Any one has faced this probem?

Sign up or log in to comment