zhifu gao
2024-01-26 4f224c88068b66bcb6f81570da59d99c9bba8288
funasr/auto/auto_model.py
@@ -88,7 +88,8 @@
class AutoModel:
    
    def __init__(self, **kwargs):
        tables.print()
        if kwargs.get("disable_log", False):
            tables.print()
        
        model, kwargs = self.build_model(**kwargs)
        
@@ -426,10 +427,10 @@
                                 f"time_escape: {time_escape_total_per_sample:0.3f}")
        end_total = time.time()
        time_escape_total_all_samples = end_total - beg_total
        print(f"rtf_avg_all: {time_escape_total_all_samples / time_speech_total_all_samples:0.3f}, "
                             f"time_speech_all: {time_speech_total_all_samples: 0.3f}, "
                             f"time_escape_all: {time_escape_total_all_samples:0.3f}")
        # end_total = time.time()
        # time_escape_total_all_samples = end_total - beg_total
        # print(f"rtf_avg_all: {time_escape_total_all_samples / time_speech_total_all_samples:0.3f}, "
        #                      f"time_speech_all: {time_speech_total_all_samples: 0.3f}, "
        #                      f"time_escape_all: {time_escape_total_all_samples:0.3f}")
        return results_ret_list