whisperX available devices #67

abdeladim-s · Aug 28, 2023 · 57bef37 · 57bef37
1 parent 854543a
commit 57bef37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/subsai/models/whisperX_model.py b/src/subsai/models/whisperX_model.py
@@ -34,8 +34,8 @@ class WhisperXModel(AbstractModel):
         'device': {
             'type': list,
             'description': "The PyTorch device to put the model into",
-            'options': ['cpu', 'cuda'],
-            'default': 'cpu'
+            'options': [None, *get_available_devices()],
+            'default': None
         },
         'compute_type': {
             'type': list,