custom-diffusion360 icon indicating copy to clipboard operation
custom-diffusion360 copied to clipboard

Sampling example error

Open luh-j opened this issue 1 year ago • 1 comments

Hello there, I was trying the sampling example:

python sample.py --custom_model_dir pretrained-models/car0 --output_dir outputs --prompt "a <new1> car beside a field of blooming sunflowers." 

And I encountered with runtime error:

RuntimeError: The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (1, 1).

I followed the instruction for setup, and I'm running it on linux ubuntu 22.04 with a5000.

luh-j avatar Sep 15 '24 02:09 luh-j

Hi @luh-j , were you able to resolve this issue?

NaveenchandraR avatar May 12 '25 10:05 NaveenchandraR