diff --git a/flake.nix b/flake.nix index d0d28c4..7934711 100644 --- a/flake.nix +++ b/flake.nix @@ -75,7 +75,14 @@ in import nixpkgs { inherit system config; - overlays = additionalOverlays ++ extOverlays; + overlays = additionalOverlays ++ extOverlays ++ [ + (final: prev: { + ctranslate2 = prev.ctranslate2.override { + withCUDA = true; + withCuDNN = true; + }; + }) + ]; }; wrapUtils = { pkgs }: diff --git a/hosts/architect/homeassistant.nix b/hosts/architect/homeassistant.nix index 99906e5..0c70049 100644 --- a/hosts/architect/homeassistant.nix +++ b/hosts/architect/homeassistant.nix @@ -15,7 +15,7 @@ in enable = true; uri = "tcp://127.0.0.1:${toString whisperPort}"; model = "large-v3"; - device = "auto"; + device = "cuda"; language = "en"; }; };