From 52264f2277a9c5cc4ae5719234c7fa578f31dd07 Mon Sep 17 00:00:00 2001
From: Guillaume Klein <guillaume.klein@systrangroup.com>
Date: Wed, 22 Mar 2023 13:51:12 +0100
Subject: [PATCH] Fix typing for device_index argument

---
 faster_whisper/transcribe.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py
index 0aeadd4..402fccd 100644
--- a/faster_whisper/transcribe.py
+++ b/faster_whisper/transcribe.py
@@ -59,7 +59,7 @@ class WhisperModel:
         self,
         model_path: str,
         device: str = "auto",
-        device_index: int = 0,
+        device_index: Union[int, List[int]] = 0,
         compute_type: str = "default",
         cpu_threads: int = 0,
         num_workers: int = 1,