Make SAII-CLDM the default pipeline

README.md

@@ -26,7 +26,7 @@ Paper: [arXiv:2506.13529](https://arxiv.org/html/2506.13529v1)
 
 ## Load From Hugging Face
 
-SAII-LDDPM:
+Default pipeline: SAII-CLDM.
 
 ```python
 import torch
@@ -40,20 +40,20 @@ pipe = DiffusionPipeline.from_pretrained(
 ).to("cuda")
 ```
 
-SAII-CLDM:
+SAII-LDDPM is also provided as a baseline:
 
 ```python
 pipe = DiffusionPipeline.from_pretrained(
     "mally-2000/saii-cldm-synthetic",
-    custom_pipeline="pipeline_cldm",
+    custom_pipeline="pipeline_lddpm",
     torch_dtype=torch.float32,
     trust_remote_code=True,
 ).to("cuda")
 ```
 
 The inference algorithms are implemented in `codes/pipeline.py`. The root
-`pipeline.py` and `pipeline_cldm.py` files are thin entry points used by
-Diffusers remote loading.
+`pipeline.py`, `pipeline_cldm.py`, and `pipeline_lddpm.py` files are thin entry
+points used by Diffusers remote loading.
 
 ## Run The Bundled Demo
 
@@ -62,8 +62,8 @@ git clone https://huggingface.co/mally-2000/saii-cldm-synthetic
 cd saii-cldm-synthetic
 pip install -r requirements.txt
 
-python infer.py       # SAII-LDDPM
 python infer.py CLDM  # SAII-CLDM
+python infer.py       # SAII-LDDPM baseline
 ```
 
 `infer.py` uses the bundled Overthrust sample and writes outputs under

model_index.json

@@ -1,5 +1,5 @@
 {
-  "_class_name": "SeismicImpInvLDDPMPipeline",
+  "_class_name": "SeismicImpInvCLDMPipeline",
   "_diffusers_version": "0.30.3",
   "condition_encoder": [
     "seismic_wavelet_encoder",
@@ -17,4 +17,4 @@
     "diffusers",
     "VQModel"
   ]
-}
+}

pipeline.py

@@ -1,3 +1,3 @@
-from .codes.pipeline import SeismicImpInvLDDPMPipeline
+from .codes.pipeline import SeismicImpInvCLDMPipeline
 
-__all__ = ["SeismicImpInvLDDPMPipeline"]
+__all__ = ["SeismicImpInvCLDMPipeline"]

pipeline_lddpm.py

@@ -0,0 +1,3 @@
+from .codes.pipeline import SeismicImpInvLDDPMPipeline
+
+__all__ = ["SeismicImpInvLDDPMPipeline"]