vram10GBか12GBかわからないけどres256まではそれなりのバッチサイズでも余裕じゃない?
自分はvram11GBだけどres384のdimsはデフォでbs6くらい
256なら16~18でも回る
ただLRDをGPUで使うと流石にOOM吐くからその時はbsを1/2に落とす