mirror of
https://github.com/gpt-omni/mini-omni
synced 2025-06-26 18:16:26 +00:00
Merge branch 'main' into main
This commit is contained in:
@@ -480,6 +480,7 @@ class OmniInference:
|
||||
with self.fabric.init_tensor():
|
||||
model.set_kv_cache(batch_size=2, device=self.device)
|
||||
|
||||
|
||||
# Load the audio and process it using Whisper
|
||||
mel, leng = load_audio(audio_path)
|
||||
audio_feature, input_ids = get_input_ids_whisper_ATBatch(mel, leng, self.whispermodel, self.device)
|
||||
|
||||
Reference in New Issue
Block a user