change where to look for pipeline

models/pipeline_loader.py

@@ -29,14 +29,14 @@ def get_pipeline(model_path):
             # Use the local pipeline file for custom_pipeline
             pipe = DiffusionPipeline.from_pretrained(
                 model_path,
-                custom_pipeline="models.text_diffusion_pipeline.TextConditionalDDPMPipeline",
+                custom_pipeline=model_path,
                 trust_remote_code=True,
             )
         else:
             # Fallback: try unconditional 
             pipe = DiffusionPipeline.from_pretrained(
                 model_path,
-                custom_pipeline="models.latent_diffusion_pipeline.UnconditionalDDPMPipeline",
+                custom_pipeline=model_path,
                 trust_remote_code=True,
             )