diff --git a/docker-compose.yml b/docker-compose.yml index c3aa30b..9fbb6d8 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -24,8 +24,17 @@ services: - driver: nvidia count: 1 capabilities: [gpu] + volumes: + # Persist Surya (datalab) + HuggingFace model caches so rebuilds don't + # re-download ~1.5 GB of weights every time. + - ix_surya_cache:/root/.cache/datalab + - ix_hf_cache:/root/.cache/huggingface labels: infrastructure.web_url: "http://192.168.68.42:8994" backup.enable: "true" backup.type: "postgres" backup.name: "infoxtractor" + +volumes: + ix_surya_cache: + ix_hf_cache: