-
-
Notifications
You must be signed in to change notification settings - Fork 203
/
docker-bake.hcl
75 lines (65 loc) · 1.55 KB
/
docker-bake.hcl
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
variable "DOCKERHUB_REPO" {
default = "timpietruskyblibla"
}
variable "DOCKERHUB_IMG" {
default = "runpod-worker-comfy"
}
variable "RELEASE_VERSION" {
default = "latest"
}
variable "HUGGINGFACE_ACCESS_TOKEN" {
default = ""
}
group "default" {
targets = ["base", "sdxl", "sd3", "flux1-schnell", "flux1-dev"]
}
target "base" {
context = "."
dockerfile = "Dockerfile"
target = "base"
platforms = ["linux/amd64"]
tags = ["${DOCKERHUB_REPO}/${DOCKERHUB_IMG}:${RELEASE_VERSION}-base"]
}
target "sdxl" {
context = "."
dockerfile = "Dockerfile"
target = "final"
args = {
MODEL_TYPE = "sdxl"
}
tags = ["${DOCKERHUB_REPO}/${DOCKERHUB_IMG}:${RELEASE_VERSION}-sdxl"]
inherits = ["base"]
}
target "sd3" {
context = "."
dockerfile = "Dockerfile"
target = "final"
args = {
MODEL_TYPE = "sd3"
HUGGINGFACE_ACCESS_TOKEN = "${HUGGINGFACE_ACCESS_TOKEN}"
}
tags = ["${DOCKERHUB_REPO}/${DOCKERHUB_IMG}:${RELEASE_VERSION}-sd3"]
inherits = ["base"]
}
target "flux1-schnell" {
context = "."
dockerfile = "Dockerfile"
target = "final"
args = {
MODEL_TYPE = "flux1-schnell"
HUGGINGFACE_ACCESS_TOKEN = "${HUGGINGFACE_ACCESS_TOKEN}"
}
tags = ["${DOCKERHUB_REPO}/${DOCKERHUB_IMG}:${RELEASE_VERSION}-flux1-schnell"]
inherits = ["base"]
}
target "flux1-dev" {
context = "."
dockerfile = "Dockerfile"
target = "final"
args = {
MODEL_TYPE = "flux1-dev"
HUGGINGFACE_ACCESS_TOKEN = "${HUGGINGFACE_ACCESS_TOKEN}"
}
tags = ["${DOCKERHUB_REPO}/${DOCKERHUB_IMG}:${RELEASE_VERSION}-flux1-dev"]
inherits = ["base"]
}