refactor: rm unused parameter and redundant return statement

facebookresearch · Oct 13, 2023 · 9af6690 · 9af6690
1 parent 5d8752d
commit 9af6690
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/audiocraft/models/multibanddiffusion.py b/audiocraft/models/multibanddiffusion.py
@@ -80,7 +80,7 @@ def get_mbd_musicgen(device=None):
         return MultiBandDiffusion(DPs=DPs, codec_model=codec_model)
 
     @staticmethod
-    def get_mbd_24khz(bw: float = 3.0, pretrained: bool = True,
+    def get_mbd_24khz(bw: float = 3.0,
                       device: tp.Optional[tp.Union[torch.device, str]] = None,
                       n_q: tp.Optional[int] = None):
         """Get the pretrained Models for MultibandDiffusion.
@@ -112,8 +112,6 @@ def get_mbd_24khz(bw: float = 3.0, pretrained: bool = True,
             DPs.append(DiffusionProcess(model=models[i], noise_schedule=schedule))
         return MultiBandDiffusion(DPs=DPs, codec_model=codec_model)
 
-        return MultiBandDiffusion(DPs, codec_model)
-
     @torch.no_grad()
     def get_condition(self, wav: torch.Tensor, sample_rate: int) -> torch.Tensor:
         """Get the conditioning (i.e. latent reprentatios of the compression model) from a waveform.