Device

Softcatala · Jan 6, 2025 · 56ead64 · 56ead64
1 parent 53bda26
commit 56ead64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
@@ -24,7 +24,7 @@ jobs:
       run: docker build -t ghcr.io/softcatala/whisper-ctranslate2:latest .
 
     - name: Use Docker image
-      run: docker run --gpus "device=0" -v "$(pwd)":/srv/files/ ghcr.io/softcatala/whisper-ctranslate2:latest /srv/files/e2e-tests/gossos.mp3
+      run: docker run -v "$(pwd)":/srv/files/ ghcr.io/softcatala/whisper-ctranslate2:latest /srv/files/e2e-tests/gossos.mp3
 
     - name: Push Docker image
       run: docker push ghcr.io/softcatala/whisper-ctranslate2:latest

diff --git a/Makefile b/Makefile
@@ -5,7 +5,7 @@ docker-build:
 	docker image ls | grep whisper-ctranslate2
 
 docker-run:
-	docker run -v "$(shell pwd)":/srv/files/ -it --rm whisper-ctranslate2 /srv/files/e2e-tests/gossos.mp3 --output_dir /srv/files/
+	docker run --gpus "device=0" -v "$(shell pwd)":/srv/files/ -it --rm whisper-ctranslate2 /srv/files/e2e-tests/gossos.mp3 --output_dir /srv/files/
 
 run:
 	python3 setup.py sdist bdist_wheel