Use aws cli to fetch model download script

guardian · Jan 31, 2025 · 9e22529 · 9e22529
1 parent 68f6a49
commit 9e22529
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/roles/whisperx/defaults/main.yml b/roles/whisperx/defaults/main.yml
@@ -0,0 +1,3 @@
+---
+amigo_data_bucket: amigo-data-prod
+model_script_stage: PROD
diff --git a/roles/whisperx/meta/main.yml b/roles/whisperx/meta/main.yml
@@ -1,6 +1,7 @@
 ---
 dependencies:
   - pip3
+  - aws-tools
   - role: packages
     packages:
       - ffmpeg
diff --git a/roles/whisperx/tasks/main.yml b/roles/whisperx/tasks/main.yml
@@ -18,12 +18,10 @@
     name: "{{ whisperx_package | default('whisperx') }}"
     executable: pip3
 
-- name: Simple GET operation
-  amazon.aws.s3_object:
-    bucket: amigo-data-prod
-    object: /deploy/PROD/whisperx-model-fetch/download_whisperx_models.py
-    dest: /tmp/download_whisperx_models.py
-    mode: get
+- name: Download models script
+  shell: |
+    aws --quiet s3 cp s3://{{ amigo_data_bucket }}/deploy/{{ model_script_stage }}/whisperx-model-fetch/download_whisperx_models.py /tmp/download_whisperx_models.py
+    exit 0
 
 - name: Download whisperx models
   command: "python3 /tmp/download_whisperx_models.py --whisper-models --diarization-models --torch-align-models --huggingface-token {{ huggingface_token }}"