Merge pull request #1606 from guardian/pm-whisperx-role

Add WhisperX Role
guardian · Jan 22, 2025 · 65deded · 65deded
2 parents 5439aa8 + 53d0f7f
commit 65deded
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/app/packer/PackerBuildConfigGenerator.scala b/app/packer/PackerBuildConfigGenerator.scala
@@ -37,8 +37,8 @@ object PackerBuildConfigGenerator {
     val instanceSize = if (requresXlargeBukder) "xlarge" else "small"
 
     val instanceType = sourceAmiMetadata.architecture match {
-      case "x86_64" => s"t3.${instanceSize}"
-      case "arm64"  => s"t4g.${instanceSize}"
+      case "x86_64" => s"t3.$instanceSize"
+      case "arm64"  => s"t4g.$instanceSize"
       case other =>
         throw new IllegalArgumentException(
           s"Don't know what instance type to use to bake an AMI for $other"

diff --git a/roles/whisperx/meta/main.yml b/roles/whisperx/meta/main.yml
@@ -0,0 +1,6 @@
+---
+dependencies:
+  - pip3
+  - role: packages
+    packages:
+      - ffmpeg
diff --git a/roles/whisperx/tasks/main.yml b/roles/whisperx/tasks/main.yml
@@ -0,0 +1,21 @@
+---
+- name: Upgrade packaging
+  pip:
+    name: "packaging"
+    state: "latest"
+    executable: pip3
+
+- name: Install whisperx dependencies
+  pip:
+    name:
+     - torch==2.0.0
+     - torchvision==0.15.1
+     - torchaudio==2.0.1
+    executable: pip3
+    extra_args: "--index-url https://download.pytorch.org/whl/cu118"
+
+- name: Install whisperx
+  pip:
+    name: whisperx
+    executable: pip3
+