From 84169d15b3c1254d260e33cd37cbc1a33255e406 Mon Sep 17 00:00:00 2001
From: FarisHijazi <theefaris@gmail.com>
Date: Fri, 2 Jun 2023 18:17:43 +0300
Subject: [PATCH] added docker-compose file

---
 README.md           |  6 ++++++
 docker-compose.yaml | 20 ++++++++++++++++++++
 2 files changed, 26 insertions(+)
 create mode 100644 docker-compose.yaml

diff --git a/README.md b/README.md
index e112f6c4f..6176788eb 100644
--- a/README.md
+++ b/README.md
@@ -55,6 +55,12 @@ accelerate launch scripts/finetune.py examples/lora-openllama-3b/config.yml \
   - `winglian/axolotl:dev`: dev branch
   - `winglian/axolotl-runpod:main`: for runpod
 
+  Or run on the current files for development:
+
+  ```sh
+  docker compose up -d
+  ```
+
 - Conda/Pip venv
   1. Install python **3.9**
 
diff --git a/docker-compose.yaml b/docker-compose.yaml
new file mode 100644
index 000000000..d40422f94
--- /dev/null
+++ b/docker-compose.yaml
@@ -0,0 +1,20 @@
+# version: '3.8'
+services:
+  axolotl:
+    build:
+      context: .
+      dockerfile: ./docker/Dockerfile
+    volumes:
+      - .:/workspace/axolotl
+      - ~/.cache/huggingface/:/root/.cache/huggingface/
+    # set environment variables
+    environment:
+      - WANDB_API_KEY=${WANDB_API_KEY}
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              # count: 1
+              capabilities: [gpu]
+    command: tail -f /dev/null