Need to call DiffusionPipeline's constructor and skip DDPM?
Browse files
models/text_diffusion_pipeline.py
CHANGED
|
@@ -2,7 +2,7 @@ import torch
|
|
| 2 |
import torch.nn.functional as F
|
| 3 |
from typing import NamedTuple, Optional
|
| 4 |
import os
|
| 5 |
-
from diffusers import DDPMPipeline, UNet2DConditionModel, DDPMScheduler
|
| 6 |
import json
|
| 7 |
# Running the main at the end of this requires messing with this import
|
| 8 |
from text_encoder.text_model import TransformerModel
|
|
|
|
| 2 |
import torch.nn.functional as F
|
| 3 |
from typing import NamedTuple, Optional
|
| 4 |
import os
|
| 5 |
+
from diffusers import DDPMPipeline, UNet2DConditionModel, DDPMScheduler, DiffusionPipeline
|
| 6 |
import json
|
| 7 |
# Running the main at the end of this requires messing with this import
|
| 8 |
from text_encoder.text_model import TransformerModel
|