Refactor sampling to use model forward pass #193
Description
Currently, the sampling method does not use the model forward pass, but instead directly accesses methods of submodules.
{{ message }}
Currently, the sampling method does not use the model forward pass, but instead directly accesses methods of submodules.