Need to call DiffusionPipeline's constructor and skip DDPM?

models/text_diffusion_pipeline.py

@@ -2,7 +2,7 @@ import torch
 import torch.nn.functional as F
 from typing import NamedTuple, Optional
 import os
-from diffusers import DDPMPipeline, UNet2DConditionModel, DDPMScheduler
+from diffusers import DDPMPipeline, UNet2DConditionModel, DDPMScheduler, DiffusionPipeline
 import json
 # Running the main at the end of this requires messing with this import
 from text_encoder.text_model import TransformerModel