Skip to content
Snippets Groups Projects
docker-compose.yaml 1.26 KiB
Newer Older
Marcin Wątroba's avatar
Marcin Wątroba committed
services:

    whisper_worker_1:
        image: gitlab.clarin-pl.eu:5050/markowanga/asr-benchmarks/whisper-worker:latest
        command: poetry run python new_datasets/whisper_asr_runner.py
        environment:
            WHISPER_MODEL: 'base'
        deploy:
            resources:
                reservations:
                    devices:
                        -   driver: nvidia
                            capabilities: [ gpu ]

    whisper_worker_2:
        image: gitlab.clarin-pl.eu:5050/markowanga/asr-benchmarks/whisper-worker:latest
        command: poetry run python new_datasets/whisper_asr_runner.py
        environment:
            WHISPER_MODEL: 'small'
        deploy:
            resources:
                reservations:
                    devices:
                        -   driver: nvidia
                            capabilities: [ gpu ]

    whisper_worker_3:
        image: gitlab.clarin-pl.eu:5050/markowanga/asr-benchmarks/whisper-worker:latest
        command: poetry run python new_datasets/whisper_asr_runner.py
        environment:
            WHISPER_MODEL: 'tiny'
        deploy:
            resources:
                reservations:
                    devices:
                        -   driver: nvidia
                            capabilities: [ gpu ]