That's weird. I cap out at 512x512 on my 16GB Ampere card. Even stepping down precision doesn't help. I wonder what's different.
I use it directly from Python.
I use it directly from Python.
I can generate a 512x512 on my 10gb 3080 no problem (or three 384x384 at a time)
It measures memory usage as well.