WhsiperX fix device type #77

abdeladim-s · Oct 29, 2023 · a6403a6 · a6403a6
1 parent 0170d61
commit a6403a6
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
@@ -33,8 +33,8 @@ class WhisperXModel(AbstractModel):
         },
         'device': {
             'type': list,
-            'description': "The PyTorch device to put the model into",
-            'options': [*get_available_devices()],
+            'description': 'Device to use for computation ("cpu", "cuda")',
+            'options': ['cpu', 'cuda'],
             'default': 'cpu'
         },
         'compute_type': {