|
|
@@ -385,11 +385,11 @@ class AutoModel:
|
|
|
result["text"] = punc_res[0]["text"]
|
|
|
|
|
|
# speaker embedding cluster after resorted
|
|
|
- if self.spk_model is not None:
|
|
|
+ if self.spk_model is not None and kwargs.get('return_spk_res', True):
|
|
|
all_segments = sorted(all_segments, key=lambda x: x[0])
|
|
|
spk_embedding = result['spk_embedding']
|
|
|
- labels = self.cb_model(spk_embedding.cpu(), oracle_num=kwargs['preset_spk_num'])
|
|
|
- del result['spk_embedding']
|
|
|
+ labels = self.cb_model(spk_embedding.cpu(), oracle_num=kwargs.get('preset_spk_num', None))
|
|
|
+ # del result['spk_embedding']
|
|
|
sv_output = postprocess(all_segments, None, labels, spk_embedding.cpu())
|
|
|
if self.spk_mode == 'vad_segment': # recover sentence_list
|
|
|
sentence_list = []
|
|
|
@@ -409,6 +409,7 @@ class AutoModel:
|
|
|
result['timestamp'], \
|
|
|
result['raw_text'])
|
|
|
result['sentence_info'] = sentence_list
|
|
|
+ del result['spk_embedding']
|
|
|
|
|
|
result["key"] = key
|
|
|
results_ret_list.append(result)
|