Move pipeline

model_index.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "TextConditionalDDPMPipeline",
   "_diffusers_version": "0.32.2",
-  "custom_pipeline": "pipeline.py",
+  "custom_pipeline": "text_diffusion_pipeline.py",
   "components": {
     "unet": { "type": "UNet2DConditionModel", "subfolder": "unet" },
     "text_encoder": { "type": "models.text_model.TransformerModel", "subfolder": "text_encoder" },

pipeline.py

@@ -1,3 +0,0 @@
-from models.text_diffusion_pipeline import TextConditionalDDPMPipeline
-
-__all__ = ["TextConditionalDDPMPipeline"]

models/text_diffusion_pipeline.py → text_diffusion_pipeline.py

@@ -10,7 +10,7 @@ import torch
 import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModel
 import util.common_settings as common_settings
-import models.sentence_transformers_helper as st_helper
+#import models.sentence_transformers_helper as st_helper
 import models.text_model as text_model
 #from models.general_training_helper import get_scene_from_embeddings