wow

modules/home/kdenlive.nix

@@ -41,13 +41,23 @@ let
   # Kdenlive's speech-to-text scripts are launched through
   # /usr/bin/env python3. Keep this environment private to Kdenlive so the
   # other Python environments installed on the system remain independent.
+  cudaPackages = pkgs.cudaPackages_13_0;
+  cudaBindings = pythonPackages.cuda-bindings.override { inherit cudaPackages; };
+  torchCuda = pythonPackages.torch-bin.override {
+    inherit cudaPackages;
+    cuda-bindings = cudaBindings;
+  };
+  whisperCuda = pythonPackages.openai-whisper.override {
+    torch = torchCuda;
+    triton = pythonPackages.triton-bin;
+  };
   kdenlivePython = pkgs.python3.withPackages (
     pythonPackages: with pythonPackages; [
       pip
       numba
-      openai-whisper
+      whisperCuda
       srt
-      torch
+      torchCuda
       vosk
     ]
   );
@@ -76,5 +86,9 @@ in
       --file "$HOME/.config/kdenliverc" \
       --group speech \
       --key speech_system_python_path ${kdenlivePython}/bin/python3
+    run ${pkgs.kdePackages.kconfig}/bin/kwriteconfig6 \
+      --file "$HOME/.config/kdenliverc" \
+      --group speech \
+      --key whisperDevice cuda
   '';
 }