custom-diffusion360
custom-diffusion360 copied to clipboard
Sampling example error
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.
Hi @luh-j , were you able to resolve this issue?