It’s odd to me that it’s failing then. The batch sizes are small, and it seems that the BEAM process is fluctuating around 5GB RAM running on the CPU.
Your GPU has 14GB available it seems, so I would expect that to work properly.
I recommend that you open an issue on GitHub because this does smell like some sort of memory leak
