Skip to content

GitLab

"tests/python/test_tokenizer.py" did not exist on "5ea40abf613e47bb56a0c06f695644d55671f585"
Commit e2678275 authored by patil-suraj's avatar patil-suraj
Browse files

add GradTTS in init

parent 1d2551d7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/diffusers/__init__.py
View file @ e2678275
...@@ -10,6 +10,6 @@ from .models.unet_glide import GLIDEUNetModel, GLIDESuperResUNetModel, GLIDEText ...@@ -10,6 +10,6 @@ from .models.unet_glide import GLIDEUNetModel, GLIDESuperResUNetModel, GLIDEText
from .models.unet_ldm import UNetLDMModel from .models.unet_ldm import UNetLDMModel
from .models.unet_grad_tts import UNetGradTTSModel from .models.unet_grad_tts import UNetGradTTSModel
from .pipeline_utils import DiffusionPipeline from .pipeline_utils import DiffusionPipeline
from .pipelines import DDIM, DDPM, GLIDE, LatentDiffusion, PNDM, BDDM from .pipelines import DDIM, DDPM, GLIDE, LatentDiffusion, PNDM, BDDM, GradTTS
from .schedulers import DDIMScheduler, DDPMScheduler, SchedulerMixin, PNDMScheduler, GradTTSScheduler from .schedulers import DDIMScheduler, DDPMScheduler, SchedulerMixin, PNDMScheduler, GradTTSScheduler
from .schedulers.classifier_free_guidance import ClassifierFreeGuidanceScheduler from .schedulers.classifier_free_guidance import ClassifierFreeGuidanceScheduler
src/diffusers/pipelines/__init__.py
View file @ e2678275
...@@ -4,3 +4,4 @@ from .pipeline_pndm import PNDM ...@@ -4,3 +4,4 @@ from .pipeline_pndm import PNDM
from .pipeline_glide import GLIDE from .pipeline_glide import GLIDE
from .pipeline_latent_diffusion import LatentDiffusion from .pipeline_latent_diffusion import LatentDiffusion
from .pipeline_bddm import BDDM from .pipeline_bddm import BDDM
from .pipeline_grad_tts import GradTTS
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment