super().__init__()
Now, we must save the unet and scheduler in a config file so that you can save your pipeline with save_pretrained.
Therefore, make sure you add every component that is save-able to the register_modules function:


	Copied
from diffusers import DiffusionPipeline
import torch


class UnetSchedulerOneForwardPipeline(DiffusionPipeline):
    def __init__(self, unet, scheduler):
        super().__init__()

        self.register_mod