CreateML StyleTransfer not working on M1 MacBook Pro

I have a MacBook Pro M1 (16 GB RAM), and testing CreateML's StyleTransfer model training.

When I press «Train», it starts processing and fails with error «Could not create buffer with format BGRA -6662».
During the «processing» it allocates about 4.5 GB of RAM.

I guess it runs out of memory, however, I've closed all other programs and I can see that there's lot of free RAM when it fails.

It happens even if I use just 3 small (500*500) images, 1 for each training, content and validation.

So, how to fix it?

I'm in the same boat here, and the error isn't very indicative of how we can solve the problem.

CreateML StyleTransfer not working on M1 MacBook Pro
 
 
Q