From 110c8a4ad205af221242db3ffdd468b5b73806fe Mon Sep 17 00:00:00 2001 From: Robert Dargavel Smith Date: Sat, 22 Apr 2023 09:35:01 +0100 Subject: [PATCH] deprecate get_input_dims --- audiodiffusion/__init__.py | 2 +- audiodiffusion/pipeline_audio_diffusion.py | 17 ----------------- 2 files changed, 1 insertion(+), 18 deletions(-) diff --git a/audiodiffusion/__init__.py b/audiodiffusion/__init__.py index 44deed2..c40f853 100644 --- a/audiodiffusion/__init__.py +++ b/audiodiffusion/__init__.py @@ -9,7 +9,7 @@ # from diffusers import AudioDiffusionPipeline from .pipeline_audio_diffusion import AudioDiffusionPipeline -VERSION = "1.5.2" +VERSION = "1.5.3" class AudioDiffusion: diff --git a/audiodiffusion/pipeline_audio_diffusion.py b/audiodiffusion/pipeline_audio_diffusion.py index 9aa80c8..dd1ed82 100644 --- a/audiodiffusion/pipeline_audio_diffusion.py +++ b/audiodiffusion/pipeline_audio_diffusion.py @@ -59,21 +59,6 @@ def __init__( super().__init__() self.register_modules(unet=unet, scheduler=scheduler, mel=mel, vqvae=vqvae) - def get_input_dims(self) -> Tuple: - """Returns dimension of input image - - Returns: - `Tuple`: (height, width) - """ - input_module = self.vqvae if self.vqvae is not None else self.unet - # For backwards compatibility - sample_size = ( - (input_module.sample_size, input_module.sample_size) - if type(input_module.sample_size) == int - else input_module.sample_size - ) - return sample_size - def get_default_steps(self) -> int: """Returns default number of steps recommended for inference @@ -131,8 +116,6 @@ def __call__( # For backwards compatibility if type(self.unet.sample_size) == int: self.unet.sample_size = (self.unet.sample_size, self.unet.sample_size) - input_dims = self.get_input_dims() - self.mel.set_resolution(x_res=input_dims[1], y_res=input_dims[0]) if noise is None: noise = torch.randn( (