Hopefully the final files needed to make model usage possible
Browse files- model_index.json +1 -0
- pipeline.py +3 -0
model_index.json
CHANGED
|
@@ -17,4 +17,5 @@
|
|
| 17 |
"diffusers",
|
| 18 |
"UNet2DConditionModel"
|
| 19 |
]
|
|
|
|
| 20 |
}
|
|
|
|
| 17 |
"diffusers",
|
| 18 |
"UNet2DConditionModel"
|
| 19 |
]
|
| 20 |
+
"pipeline": "TextConditionalDDPMPipeline"
|
| 21 |
}
|
pipeline.py
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from models.text_diffusion_pipeline import TextConditionalDDPMPipeline
|
| 2 |
+
|
| 3 |
+
__all__ = ["TextConditionalDDPMPipeline"]
|