diff --git a/docker-compose.yml b/docker-compose.yml
index a589feed..bdf86aed 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,6 +11,7 @@ services:
       - ./aitk_db.db:/app/ai-toolkit/aitk_db.db
       - ./datasets:/app/ai-toolkit/datasets
       - ./output:/app/ai-toolkit/output
+      - ./config:/app/ai-toolkit/config
     environment:
       - AI_TOOLKIT_AUTH=${AI_TOOLKIT_AUTH:-password}
       - NODE_ENV=production
diff --git a/requirements.txt b/requirements.txt
index b2ab0286..12c9ce89 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 torch==2.6.0
 torchvision==0.21.0
+torchao==0.9.0
 safetensors
 git+https://github.com/huggingface/diffusers@363d1ab7e24c5ed6c190abb00df66d9edb74383b
 transformers==4.49.0