Cleanup. (#9838)

Author: comfyanonymous

comfy/audio_encoders/audio_encoders.py

@@ -31,7 +31,7 @@ def get_sd(self):
     def encode_audio(self, audio, sample_rate):
         comfy.model_management.load_model_gpu(self.patcher)
         audio = torchaudio.functional.resample(audio, sample_rate, self.model_sample_rate)
-        out, all_layers = self.model(audio.to(self.load_device), sr=self.model_sample_rate)
+        out, all_layers = self.model(audio.to(self.load_device))
         outputs = {}
         outputs["encoded_audio"] = out
         outputs["encoded_audio_all_layers"] = all_layers

comfy/audio_encoders/wav2vec2.py

@@ -238,7 +238,7 @@ def __init__(
             device=device, dtype=dtype, operations=operations
         )
 
-    def forward(self, x, sr=16000, mask_time_indices=None, return_dict=False):
+    def forward(self, x, mask_time_indices=None, return_dict=False):
         x = torch.mean(x, dim=1)
 
         if self.do_normalize: