n reward.


unet (UNet1DModel) — U-Net architecture to denoise the encoded trajectories.


scheduler (SchedulerMixin) —
A scheduler to be used in combination with unet to denoise the encoded trajectories. Default for this
application is DDPMScheduler.
env — An environment following the OpenAI gym API to act in. For now only Hopper has pretrained models.



This model inherits from DiffusionPipeline. Check the superclass documentation for the ge