Rename pipelines (huggingface#115)

up

Author: patrickvonplaten

__init__.py

@@ -9,11 +9,11 @@
 from .modeling_utils import ModelMixin
 from .models import AutoencoderKL, UNet2DConditionModel, UNet2DModel, VQModel
 from .pipeline_utils import DiffusionPipeline
-from .pipelines import DDIMPipeline, DDPMPipeline, LatentDiffusionUncondPipeline, PNDMPipeline, ScoreSdeVePipeline
+from .pipelines import DDIMPipeline, DDPMPipeline, LDMPipeline, PNDMPipeline, ScoreSdeVePipeline
 from .schedulers import DDIMScheduler, DDPMScheduler, PNDMScheduler, SchedulerMixin, ScoreSdeVeScheduler
 
 
 if is_transformers_available():
-    from .pipelines import LatentDiffusionPipeline
+    from .pipelines import LDMTextToImagePipeline
 else:
     from .utils.dummy_transformers_objects import *

pipelines/__init__.py

@@ -1,10 +1,10 @@
 from ..utils import is_inflect_available, is_transformers_available, is_unidecode_available
 from .ddim import DDIMPipeline
 from .ddpm import DDPMPipeline
-from .latent_diffusion_uncond import LatentDiffusionUncondPipeline
+from .latent_diffusion_uncond import LDMPipeline
 from .pndm import PNDMPipeline
 from .score_sde_ve import ScoreSdeVePipeline
 
 
 if is_transformers_available():
-    from .latent_diffusion import LatentDiffusionPipeline
+    from .latent_diffusion import LDMTextToImagePipeline

pipelines/latent_diffusion/__init__.py

@@ -2,4 +2,4 @@
 
 
 if is_transformers_available():
-    from .pipeline_latent_diffusion import LatentDiffusionPipeline, LDMBertModel
+    from .pipeline_latent_diffusion import LDMBertModel, LDMTextToImagePipeline

pipelines/latent_diffusion/pipeline_latent_diffusion.py

@@ -14,7 +14,7 @@
 from ...pipeline_utils import DiffusionPipeline
 
 
-class LatentDiffusionPipeline(DiffusionPipeline):
+class LDMTextToImagePipeline(DiffusionPipeline):
     def __init__(self, vqvae, bert, tokenizer, unet, scheduler):
         super().__init__()
         scheduler = scheduler.set_format("pt")

pipelines/latent_diffusion_uncond/__init__.py

@@ -1 +1 @@
-from .pipeline_latent_diffusion_uncond import LatentDiffusionUncondPipeline
+from .pipeline_latent_diffusion_uncond import LDMPipeline

pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py

@@ -5,7 +5,7 @@
 from ...pipeline_utils import DiffusionPipeline
 
 
-class LatentDiffusionUncondPipeline(DiffusionPipeline):
+class LDMPipeline(DiffusionPipeline):
     def __init__(self, vqvae, unet, scheduler):
         super().__init__()
         scheduler = scheduler.set_format("pt")

utils/dummy_transformers_objects.py

@@ -3,42 +3,7 @@
 from ..utils import DummyObject, requires_backends
 
 
-class GlideSuperResUNetModel(metaclass=DummyObject):
-    _backends = ["transformers"]
-
-    def __init__(self, *args, **kwargs):
-        requires_backends(self, ["transformers"])
-
-
-class GlideTextToImageUNetModel(metaclass=DummyObject):
-    _backends = ["transformers"]
-
-    def __init__(self, *args, **kwargs):
-        requires_backends(self, ["transformers"])
-
-
-class GlideUNetModel(metaclass=DummyObject):
-    _backends = ["transformers"]
-
-    def __init__(self, *args, **kwargs):
-        requires_backends(self, ["transformers"])
-
-
-class UNetGradTTSModel(metaclass=DummyObject):
-    _backends = ["transformers"]
-
-    def __init__(self, *args, **kwargs):
-        requires_backends(self, ["transformers"])
-
-
-class GlidePipeline(metaclass=DummyObject):
-    _backends = ["transformers"]
-
-    def __init__(self, *args, **kwargs):
-        requires_backends(self, ["transformers"])
-
-
-class LatentDiffusionPipeline(metaclass=DummyObject):
+class LDMTextToImagePipeline(metaclass=DummyObject):
     _backends = ["transformers"]
 
     def __init__(self, *args, **kwargs):