Update openvoice/se_extractor.py
Browse files
openvoice/se_extractor.py
CHANGED
@@ -19,7 +19,7 @@ model = None
|
|
19 |
def split_audio_whisper(audio_path, audio_name, target_dir='processed'):
|
20 |
global model
|
21 |
if model is None:
|
22 |
-
model = WhisperModel(model_size, device='cuda' if torch.cuda.is_available() else 'cpu', compute_type="
|
23 |
audio = AudioSegment.from_file(audio_path)
|
24 |
max_len = len(audio)
|
25 |
|
|
|
19 |
def split_audio_whisper(audio_path, audio_name, target_dir='processed'):
|
20 |
global model
|
21 |
if model is None:
|
22 |
+
model = WhisperModel(model_size, device='cuda' if torch.cuda.is_available() else 'cpu', compute_type="float8")
|
23 |
audio = AudioSegment.from_file(audio_path)
|
24 |
max_len = len(audio)
|
25 |
|