Mask_generation_function loading ...