Prevents device_index to read after its argument preventing for examp…

…le to go before the filename. We only support one device ID
Softcatala · Dec 29, 2023 · e4e3532 · e4e3532
1 parent c1d6e0f
commit e4e3532
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/whisper_ctranslate2/whisper_ctranslate2.py b/src/whisper_ctranslate2/whisper_ctranslate2.py
@@ -165,10 +165,9 @@ def read_command_line():
 
     computing_args.add_argument(
         "--device_index",
-        nargs="*",
         type=int,
         default=0,
-        help="device IDs where to place this model on",
+        help="device ID where to place this model on",
     )
 
     computing_args.add_argument(