<enhance>: support using float16 in inference to speed up

2026-02-05 18:09:19 +08:00 · 2024-04-27 14:26:50 +08:00
parent 2c52de01b4
commit 865a68c60e
6 changed files with 103 additions and 51 deletions
--- a/musetalk/whisper/audio2feature.py
+++ b/musetalk/whisper/audio2feature.py
@@ -13,7 +13,11 @@ class Audio2Feature():
        self.whisper_model_type = whisper_model_type
        self.model = load_model(model_path) #

-    def get_sliced_feature(self,feature_array, vid_idx, audio_feat_length= [2,2],fps = 25):
+    def get_sliced_feature(self,
+                           feature_array, 
+                           vid_idx, 
+                           audio_feat_length=[2,2],
+                           fps=25):
        """
        Get sliced features based on a given index
        :param feature_array: