import encoder from elsewhere

text_diffusion_pipeline.py

@@ -5,13 +5,13 @@ import os
 from diffusers import DDPMPipeline, UNet2DConditionModel, DDPMScheduler
 import json
 # Running the main at the end of this requires messing with this import
-from text_model import TransformerModel  
+from text_encoder.text_model import TransformerModel  
 import torch
 import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModel
 import common_settings as common_settings
 import sentence_transformers_helper as st_helper
-import text_model as text_model
+import text_encoder.text_model as text_model
 from general_training_helper import get_scene_from_embeddings
             
 class PipelineOutput(NamedTuple):