Samples from the generative model (orange curves) for the inversion of six Stokes profiles extracted from the test set. The blue curve shows the original model, the green curve shows the median model, and the red curve shows the greedy decoded model. The upper three rows show the results when the Fe I spectral region was used, the middle three rows show the results when the Ca II spectral region was used, and the last three rows show the results when both spectral regions were used.