Skip to content

Commit

Permalink
set hf cache to be within the storage/models dir
Browse files Browse the repository at this point in the history
  • Loading branch information
amitkparekh committed Dec 5, 2023
1 parent 8ea8983 commit 274bb2c
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion docker/docker-compose.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,21 +9,27 @@ x-healthcheck-defaults: &healthcheck-defaults
retries: 3
start_period: 30s

x-environment-defaults: &environment-defaults
HUGGINGFACE_HUB_CACHE: /app/model/huggingface/hub
HUGGINGFACE_ASSETS_CACHE: /app/model/huggingface/assets
HUGGINGFACE_HUB_VERBOSITY: info

services:
feature_extractor:
container_name: feature_extractor
image: ghcr.io/emma-heriot-watt/perception:cpu
environment:
LOG_LEVEL: "debug"
CLASSMAP_TYPE: "simbot"
<<: *environment-defaults
ports:
- "5500:5500"
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:5500/ping"]
<<: *healthcheck-defaults
volumes: *model-volume
entrypoint: python
command: src/emma_perception/commands/run_server.py --config_file "src/emma_perception/constants/vinvl_x152c4_simbot_customised.yaml"
command: src/emma_perception/commands/run_server.py

intent_extractor:
container_name: intent_extractor
Expand All @@ -33,6 +39,7 @@ services:
MODEL_NAME: "heriot-watt/emma-base"
MODEL_CHECKPOINT_PATH: "/app/model/${INTENT_EXTRACTOR_MODEL}"
ENABLE_PREDICTION_PATCHING: False
<<: *environment-defaults
ports:
- "5501:6000"
healthcheck:
Expand All @@ -49,6 +56,7 @@ services:
LOG_LEVEL: debug
MODEL_CHECKPOINT_PATH: "/app/model/${INSTRUCTION_PREDICTOR_MODEL}"
ENABLE_PREDICTION_PATCHING: False
<<: *environment-defaults
ports:
- "5502:6000"
healthcheck:
Expand Down

0 comments on commit 274bb2c

Please sign in to comment.