Skip to content

Commit 7986943

Browse files
author
DHBaek
committed
refactor: update envs and docker compose
1 parent 10c1a48 commit 7986943

File tree

3 files changed

+6
-25
lines changed

3 files changed

+6
-25
lines changed

README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ docker build -t tti-worker .
4646
```shell
4747
docker run -d --name <worker_container_name> \
4848
--gpus='"device=0"' -e BROKER_URI=<broker_uri> \
49-
-e DATABASE_URL=<firebase_realtime_database_url> \
50-
-e STORAGE_BUCKET=<firebase_storage_url> \
49+
-e FIREBASE_DATABASE_URL=<firebase_realtime_database_url> \
50+
-e FIREBASE_STORAGE_BUCKET=<firebase_storage_url> \
5151
-v <firebase_credential_path>:/app/key -v <model_local_path>:/app/model \
5252
tti-worker
5353
```

docker-compose.yml

+3-22
Original file line numberDiff line numberDiff line change
@@ -1,29 +1,10 @@
11
version: '3'
22

33
services:
4-
stable-diffusion-v2:
5-
image: devainetwork/tti-worker
6-
container_name: stable-diffuion-v2-worker
7-
env_file:
8-
- .env
9-
environment:
10-
- VHOST_NAME=stable-diffusion-v2
11-
volumes:
12-
- <firebase_credential_path>:/app/key
13-
- <model_local_path>:/app/model
14-
networks:
15-
- tti-fastapi_dev
16-
restart: unless-stopped
17-
deploy:
18-
resources:
19-
reservations:
20-
devices:
21-
- driver: nvidia
22-
device_ids: [<gpu_id>]
23-
capabilities: [gpu]
24-
254
stable-diffusion-v2-1:
26-
image: devainetwork/tti-worker
5+
build:
6+
context: .
7+
dockerfile: Dockerfile
278
container_name: stable-diffuion-v2-1-worker
289
env_file:
2910
- .env

src/configs/config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ class FirebaseSettings(BaseSettings):
1717
cred_path: str = "/app/key/serviceAccountKey.json"
1818
firebase_database_url: str
1919
firebase_storage_bucket: str
20-
app_name: str = "text-to-art"
20+
firebase_app_name: str = "text-to-art"
2121

2222

2323
celery_worker_settings = CeleryWorkerSettings()

0 commit comments

Comments
 (0)