schrum2 commited on
Commit
5ffad80
·
1 Parent(s): 0ef1dd5

Access pipeline via other file

Browse files
Files changed (1) hide show
  1. model_index.json +1 -1
model_index.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "TextConditionalDDPMPipeline",
3
  "_diffusers_version": "0.32.2",
4
- "custom_pipeline": "models/text_diffusion_pipeline.py",
5
  "components": {
6
  "unet": { "type": "UNet2DConditionModel", "subfolder": "unet" },
7
  "text_encoder": { "type": "models.text_model.TransformerModel", "subfolder": "text_encoder" },
 
1
  {
2
  "_class_name": "TextConditionalDDPMPipeline",
3
  "_diffusers_version": "0.32.2",
4
+ "custom_pipeline": "pipeline.py",
5
  "components": {
6
  "unet": { "type": "UNet2DConditionModel", "subfolder": "unet" },
7
  "text_encoder": { "type": "models.text_model.TransformerModel", "subfolder": "text_encoder" },