-
Notifications
You must be signed in to change notification settings - Fork 8
/
docker-compose.yaml
159 lines (145 loc) · 3.99 KB
/
docker-compose.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
version: '3.5'
services:
mongo:
image: 'mongo:latest'
network_mode: host
redis:
image: 'redis:latest'
network_mode: host
esmfold:
image: 'ghcr.io/basedlabs/esmfold:2.0.0'
network_mode: host
build:
context: microservices/esmfold
dockerfile: build/Dockerfile
command: python worker.py
env_file:
- microservices/esmfold/service/.env
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [ gpu ]
arxiv-ai-abstractions-search:
image: 'ghcr.io/basedlabs/arxiv-ai-abstractions-search:1.0.0'
network_mode: host
command: python worker.py
build:
context: microservices/arxiv_abstracts
dockerfile: build/Dockerfile
env_file:
- microservices/arxiv_abstracts/service/.env
volumes:
- ${ARXIV_ABSTRACTS_DB}:/app/chroma_db
esmfold-light:
image: 'ghcr.io/basedlabs/esmfold-light:2.0.0'
network_mode: host
command: python worker.py
build:
context: microservices/esmfold_light
dockerfile: build/Dockerfile
env_file:
- microservices/esmfold_light/service/.env
blast-query:
image: 'ghcr.io/basedlabs/blast-query:1.0.0'
network_mode: host
command: python worker.py
build:
context: microservices/blast_query
dockerfile: build/Dockerfile
env_file:
- microservices/blast_query/service/.env
rfdiffusion:
image: 'ghcr.io/basedlabs/rfdiffusion:2.0.0'
network_mode: host
build:
context: microservices/rfdiffusion
dockerfile: build/Dockerfile
command: python3.9 worker.py
env_file:
- microservices/rfdiffusion/service/.env
volumes:
- ${RFDIFFUSION_WEIGHTS_LOCATION}:/app/RFdiffusion/models
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [ gpu ]
diffdock:
image: 'ghcr.io/basedlabs/diffdock:2.0.0'
network_mode: host
build:
context: microservices/diffdock
dockerfile: build/Dockerfile
command: python worker.py
env_file:
- microservices/diffdock/service/.env
volumes:
- ${DIFFDOCK_WEIGHTS_LOCATION}:/app/DiffDock/esm/model_weights/.cache/torch/hub/checkpoints
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [ gpu ]
external-data-query:
image: 'ghcr.io/basedlabs/external-data-query:1.1.3'
network_mode: host
build:
context: microservices/external_data_query
dockerfile: build/Dockerfile
command: --host=0.0.0.0 --port=5739
biobuddy:
image: 'ghcr.io/basedlabs/biobuddy:1.3.1'
network_mode: host
build:
context: microservices/biobuddy
dockerfile: build/Dockerfile
env_file:
- microservices/diffdock/service/.env
command: --host=0.0.0.0 --port=5738
proteinmpnn:
image: 'ghcr.io/basedlabs/proteinmpnn:1.0.0'
network_mode: host
build:
context: microservices/proteinmpnn
dockerfile: build/Dockerfile
env_file:
- microservices/proteinmpnn/service/.env
command: python worker.py
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [ gpu ]
nolabs-backend:
image: 'ghcr.io/basedlabs/nolabs-backend:1.1.0'
network_mode: host
working_dir: /app
build:
context: .
dockerfile: build/backend.Dockerfile
command: python nolabs/application/api.py
environment:
- PYTHONPATH=/app
env_file:
- nolabs/infrastructure/.env
nolabs-worker:
extends:
service: nolabs-backend
command: python nolabs/workflow/worker.py
nolabs-frontend:
image: 'ghcr.io/basedlabs/nolabs-frontend:1.1.0'
network_mode: host
working_dir: /app
build:
context: .
dockerfile: build/frontend.Dockerfile
command: npm --prefix frontend run dev -- --host