diff --git a/docker-compose.yml b/docker-compose.yml index 7476d03..d034f38 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -51,7 +51,7 @@ services: # `--profile gpu`. Uses the psyc-trainer image (built from Dockerfile.train). inference: image: psyc-trainer - command: ["/scripts/serve_model.py", "--adapter", "/data/adapters/psyc-v4/final", "--host", "0.0.0.0", "--port", "8771"] + command: ["/scripts/serve_model.py", "--adapter", "/data/adapters/psyc-v5/final", "--host", "0.0.0.0", "--port", "8771"] volumes: - ./data:/data - ./scripts:/scripts