Spaces:
Sleeping
Sleeping
Manjot Singh
commited on
Commit
·
8041c04
1
Parent(s):
3000240
change dtype
Browse files- audio_processing.py +1 -1
audio_processing.py
CHANGED
|
@@ -35,7 +35,7 @@ def process_audio(audio_file, translate=False, model_size="small"):
|
|
| 35 |
try:
|
| 36 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 37 |
print(f"Using device: {device}")
|
| 38 |
-
compute_type = "
|
| 39 |
audio = whisperx.load_audio(audio_file)
|
| 40 |
model = whisperx.load_model(model_size, device, compute_type=compute_type)
|
| 41 |
|
|
|
|
| 35 |
try:
|
| 36 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 37 |
print(f"Using device: {device}")
|
| 38 |
+
compute_type = "float16"
|
| 39 |
audio = whisperx.load_audio(audio_file)
|
| 40 |
model = whisperx.load_model(model_size, device, compute_type=compute_type)
|
| 41 |
|