Create ML Style Transfer - Reduce model size

Hello,
I'm creating models using Create ML with the new Style Transfer.

In the WWDC video it can be seen that the size of the model can be less than 1MB.

Each model that I try to create is 6.7MB of output size, even changing the parameters and sources to train the models, the size remains high and always identical.

I would like to know if it is a beta bug or if there are parameters that can reduce the size of the models to be imported into Xcode.

Thanks