gpu memory about maisi size 512 512 256

[CugAgo]

when I use the /maisi/scripts/diff_model_create_training_data.py to create my data, if my image size is 512512256, the memory will be too large, and if size is 512512128, it can run successfully.