Merge remote-tracking branch 'origin/duplex-stt-rebased' into duplex-stt-rebased-latest

Author: kevinhu-nv

nemo/collections/speechlm2/models/duplex_stt_model.py

@@ -16,6 +16,7 @@
 import random
 import tempfile
 
+import soundfile as sf
 import torch
 import torch.distributed as dist
 import torch.nn.functional as F
@@ -176,9 +177,7 @@ def load_pretrained_s2s_model(self, checkpoint_path):
             loaded_keys = []
             missing_keys = []
 
-            with safe_open(
-                os.path.join(checkpoint_path, "model.safetensors"), framework="pt", device="cpu"
-            ) as f:
+            with safe_open(os.path.join(checkpoint_path, "model.safetensors"), framework="pt", device="cpu") as f:
                 available_keys = f.keys()
                 for key in available_keys:
                     if key in model_state_dict: