From 223815060f694b54666f1bd7860ac027ccd5db00 Mon Sep 17 00:00:00 2001
From: Marko Henning <marko.henning@posteo.de>
Date: Wed, 27 Nov 2024 14:11:15 +0100
Subject: [PATCH] Add device cli parameter to model kwargs.

---
 scraibe/cli.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scraibe/cli.py b/scraibe/cli.py
index e4eeaad..323a1b1 100644
--- a/scraibe/cli.py
+++ b/scraibe/cli.py
@@ -100,6 +100,7 @@ def str2bool(string):
                     'whisper_type':arg_dict.pop("whisper_type"),
                     'dia_model': arg_dict.pop("diarization_directory"),
                     'use_auth_token': arg_dict.pop("hf_token"),
+                    'device': arg_dict.pop('inference_device')
                     }
 
     if arg_dict["whisper_model_directory"]: