# originally generative-models/scripts, for sampling stuffs