fix: return latent vars if not in the training stage;

fix: return latent vars if not in the training stage; #250