Merge pull request #20 from ainize-team/feature/donghoon/use-docker-compose

Feature/donghoon/use docker compose

Author: DHBaek

.env.sample

@@ -1,5 +1,4 @@
-BROKER_BASE_URI=amqp://{user-name}:{password}@172.17.0.1:5674
-VHOST_NAME=stable-diffusion-v2
-DATABASE_URL={filebase-database-url}
-STORAGE_BUCKET={filebase-storage-bucket}
+BROKER_BASE_URI=amqp://<user_name>:<password>@<rabbitmq_container_name>:5672
+DATABASE_URL=<filebase_database_url>
+STORAGE_BUCKET=<filebase_storage_bucket>
 APP_NAME=text-to-art

.gitignore

@@ -128,4 +128,7 @@ dmypy.json
 # Pyre type checker
 .pyre/
 
-env.list
+env.list
+
+# FireBase
+serviceAccountKey.json

README.md

@@ -2,35 +2,47 @@
 
 Serving Text To Image Model Using FastAPI and Celery
 
-## For Developers
 
-1. install dev package.
 
+## How to start
+### Using docker-compose(recommended)
+1. Create RabbitMQ and FastAPI container(refer to [TTI-FastAPI](https://github.com/ainize-team/TTI-FastAPI))
+
+2. Clone repository
 ```shell
-pip install -r requirements.txt
-pip install -r requirements-dev.txt
+git clone https://github.com/ainize-team/TTI-Worker
+cd TTI-Worker
 ```
 
-2. install pre-commit.
+3. Edit [docker-compose.yml](./docker-compose.yml) and [.env file](./.env.sample) for your project.
+- Vhost for each worker is in [docker-compose.yml](./docker-compose.yml) and common RabbitMQ config is in [.env file](./.env.sample).
 
+4. Run worker container
 ```shell
-pre-commit install
+docker-compose up -d
+
+# If you want to run a specific worker container, write service name.
+docker-compose up -d <service name>
 ```
 
-## Installation
-1. Run RabbitMQ image as a broker
+### Using docker
+1. Run RabbitMQ comtainer as a broker
 ```shell
-docker run -d --name tti-rabbitmq -p 5672:5672 -p 8080:15672 --restart=unless-stopped rabbitmq:3.9.21-management
+docker run -d --name tti-rabbitmq -p 5672:5672 -p 15672:15672 --restart=unless-stopped rabbitmq:3.11.2-management
 ```
 
-2. Build docker image
+2. Clone repository
 ```shell
-git clone https://github.com/ainize-team/TTI-Worker.git
+git clone https://github.com/ainize-team/TTI-Worker
 cd TTI-Worker
+```
+
+3. Build docker image
+```shell
 docker build -t tti-worker .
 ```
 
-3. Run docker image
+4. Create docker container
 ```shell
 docker run -d --name <worker_container_name> \
 --gpus='"device=0"' -e BROKER_URI=<broker_uri> \
@@ -50,3 +62,19 @@ tti-worker
 
 ## Test with FastAPI
 - Check our [TTI-FastAPI](https://github.com/ainize-team/TTI-FastAPI) Repo.
+
+
+## For Developers
+
+1. Install dev package
+
+```shell
+pip install -r requirements.txt
+pip install -r requirements-dev.txt
+```
+
+2. Install pre-commit
+
+```shell
+pre-commit install
+```

docker-compose.yml

@@ -0,0 +1,52 @@
+version: '3'
+
+services:
+  stable-diffusion-v2:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    container_name: stable-diffuion-v2-worker
+    env_file:
+      - .env
+    environment:
+      - VHOST_NAME=stable-diffusion-v2
+    volumes:
+      - <firebase_credential_path>:/app/key
+      - <model_local_path>:/app/model
+    networks:
+      - tti-fastapi_dev
+    restart: unless-stopped
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: [<gpu_id>]
+              capabilities: [gpu]
+
+  stable-diffusion-v2-1:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    container_name: stable-diffuion-v2-1-worker
+    env_file:
+      - .env
+    environment:
+      - VHOST_NAME=stable-diffusion-v2-1
+    volumes:
+      - <firebase_credential_path>:/app/key
+      - <model_local_path>:/app/model
+    networks:
+      - tti-fastapi_dev
+    restart: unless-stopped
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: [<gpu_id>]
+              capabilities: [gpu]
+
+networks:
+  tti-fastapi_dev:
+    external: true