Using model_set_axis to avoid looping on wavelength planes (same recipe as create_psf_cube).
model_set_axis