Add device cli parameter to model kwargs.

JSchmie · Nov 27, 2024 · 2238150 · 2238150
1 parent d00ec2d
commit 2238150
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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"]: