| services: | |
| subgen: | |
| container_name: subgen | |
| tty: true | |
| image: mccloud/subgen | |
| environment: | |
| - "WHISPER_MODEL=medium" | |
| - "WHISPER_THREADS=4" | |
| - "PROCADDEDMEDIA=True" | |
| - "PROCMEDIAONPLAY=False" | |
| - "NAMESUBLANG=aa" | |
| - "SKIPIFINTERNALSUBLANG=eng" | |
| - "PLEXTOKEN=plextoken" | |
| - "PLEXSERVER=http://plexserver:32400" | |
| - "JELLYFINTOKEN=token here" | |
| - "JELLYFINSERVER=http://jellyfin:8096" | |
| - "WEBHOOKPORT=9000" | |
| - "CONCURRENT_TRANSCRIPTIONS=2" | |
| - "WORD_LEVEL_HIGHLIGHT=False" | |
| - "DEBUG=True" | |
| - "USE_PATH_MAPPING=False" | |
| - "PATH_MAPPING_FROM=/tv" | |
| - "PATH_MAPPING_TO=/Volumes/TV" | |
| - "TRANSCRIBE_DEVICE=cpu" | |
| - "CLEAR_VRAM_ON_COMPLETE=True" | |
| - "MODEL_PATH=./models" | |
| - "UPDATE=False" | |
| - "APPEND=False" | |
| - "USE_MODEL_PROMPT=False" | |
| - "CUSTOM_MODEL_PROMPT=" | |
| - "LRC_FOR_AUDIO_FILES=True" | |
| - "CUSTOM_REGROUP=cm_sl=84_sl=42++++++1" | |
| volumes: | |
| - "${TV}:/tv" | |
| - "${MOVIES}:/movies" | |
| - "${APPDATA}/subgen/models:/subgen/models" | |
| ports: | |
| - "9000:9000" | |
| # # ⚠️ FOR Nvidia GPU IMAGE ⚠️ | |
| # # uncomment folllowing lines to pass nvidia gpu to the container | |
| # deploy: | |
| # resources: | |
| # reservations: | |
| # devices: | |
| # - driver: nvidia | |
| # count: 1 | |
| # capabilities: | |
| # - gpu | |
| # | |
| # | |
| # # ⚠️ For Native Linux AMD GPU IMAGE ⚠️ | |
| # # Change image to mccloud/subgen:amd and TRANSCRIBE_DEVICE to cuda, then | |
| # # uncomment the following lines to pass an AMD GPU (ROCm) to the container. | |
| # # | |
| # # CT2_CUDA_ALLOCATOR fixes GPU page faults on RDNA3 (gfx1100/1101) — OpenNMT/CTranslate2#2021 | |
| # # HSA_OVERRIDE_GFX_VERSION may be needed for newer RDNA GPUs not yet in ROCm's default target list. | |
| # environment: | |
| # - "CT2_CUDA_ALLOCATOR=cub_caching" | |
| # - "HSA_OVERRIDE_GFX_VERSION=11.0.0" # adjust to your gfx version (e.g. 11.0.0 for RX 7900 series) | |
| # - "TRANSCRIBE_DEVICE=cuda" | |
| # devices: | |
| # - /dev/kfd | |
| # - /dev/dri | |
| # group_add: | |
| # - video | |
| # - render | |
| # | |
| # | |
| # # ⚠️ For WSL AMD GPU IMAGE ⚠️ | |
| # # Requirements | |
| # # WSL ubuntu 22.04 matching the CTranslate2 container base os running via WSL 2. | |
| # # Librocmdxg https://github.com/ROCm/librocdxg | |
| # # Change image to mccloud/subgen:amd and TRANSCRIBE_DEVICE to cuda, then | |
| # # uncomment the following lines to pass an AMD GPU (ROCm) to the container. | |
| # environment: | |
| # - "HSA_ENABLE_DXG_DETECTION=1" | |
| # - "TRANSCRIBE_DEVICE=cuda" | |
| # devices: | |
| # - /dev/dxg | |
| # volumes: | |
| # - /usr/lib/wsl/lib/libdxcore.so:/lib/libdxcore.so | |
| # - /opt/rocm/lib/librocdxg.so:/lib/librocdxg.so | |