-
-
Notifications
You must be signed in to change notification settings - Fork 7
Expand file tree
/
Copy pathbuild.env.example
More file actions
129 lines (117 loc) · 4.1 KB
/
Copy pathbuild.env.example
File metadata and controls
129 lines (117 loc) · 4.1 KB
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
# Build-time configuration (Dockerfile ARGs)
# Copy to `build.env` if you want, then export vars before building.
#
# Notes:
# - These variables only affect image builds (Dockerfile `ARG`), not runtime.
# - They are NOT read by `docker-compose.yml` in this repo because it uses a prebuilt image.
#
# Build examples:
# - Makefile: `make build CUDA_PROFILE=cu130 INSTALL_GPU_STACK=1 INSTALL_INVOKE=1 INSTALL_AI_TOOLKIT=1`
# - Docker: `docker build --build-arg CUDA_PROFILE=cu130 --build-arg CUDA_BASE_IMAGE=nvidia/cuda:13.0.2-runtime-ubuntu22.04 --build-arg TORCH_VERSION=2.11.0 -t lora-pilot:dev .`
# ========================================
# CUDA PROFILE
# ========================================
# Default Blackwell profile. Use cu128 values below for the legacy CUDA 12.8 build.
CUDA_PROFILE=cu130
CUDA_BASE_IMAGE=nvidia/cuda:13.0.2-runtime-ubuntu22.04
# ========================================
# COMPONENT INSTALLATION (0/1)
# ========================================
INSTALL_GPU_STACK=1
INSTALL_COMFY=1
INSTALL_KOHYA=1
INSTALL_INVOKE=1
INSTALL_DIFFPIPE=1
INSTALL_AI_TOOLKIT=1
INSTALL_AI_TOOLKIT_UI=1
INSTALL_COPILOT_CLI=1
# ========================================
# COPILOT CLI
# ========================================
# Default pinned to the latest published Copilot CLI package version.
COPILOT_CLI_VERSION=1.0.10
# ========================================
# DEV TOOLS
# ========================================
NODE_MAJOR=24
CODE_SERVER_VERSION=4.127.0
NPM_VERSION=11.18.0
JUPYTERLAB_VERSION=4.6.1
IPYWIDGETS_VERSION=8.1.8
# ========================================
# UPSTREAM REFS
# ========================================
# Defaults are pinned to verified upstream refs instead of floating on repo HEAD.
COMFYUI_REF=v0.27.0
# ComfyUI's built-in Manager package version, used with --enable-manager.
COMFYUI_MANAGER_REF=4.2.2
COMFYUI_DOWNLOADER_REF=03146df738191004a8aad8264dca5c3530907f56
KOHYA_REF=v25.2.1
DIFFPIPE_REF=a7e7decf4325c1f03e4b88b7de93640029abd011
AI_TOOLKIT_REF=6c0d1c4679cf8fe153ef56bdc779c93239e1cf0f
# ========================================
# TRAINING STACKS
# ========================================
# AI Toolkit keeps its own venv but imports the shared core GPU stack.
AI_TOOLKIT_DIFFUSERS_VERSION=git
DIFFPIPE_DIFFUSERS_VERSION=0.38.0
DIFFPIPE_TRANSFORMERS_VERSION=5.11.0
TENSORBOARD_VERSION=2.21.0
# ========================================
# VERSION PINS (GPU STACK)
# ========================================
TORCH_VERSION=2.11.0
TORCHVISION_VERSION=0.26.0
TORCHAUDIO_VERSION=2.11.0
TORCH_INDEX_URL=https://download.pytorch.org/whl/cu130
# CUDA 12.8 legacy profile:
# CUDA_PROFILE=cu128
# CUDA_BASE_IMAGE=nvidia/cuda:12.8.1-runtime-ubuntu22.04
# TORCH_VERSION=2.11.0
# TORCHVISION_VERSION=0.26.0
# TORCHAUDIO_VERSION=2.11.0
# TORCH_INDEX_URL=https://download.pytorch.org/whl/cu128
# CUDA_NVCC_PKG=cuda-nvcc-12-8
XFORMERS_VERSION=0.0.35
BITSANDBYTES_VERSION=0.49.2
CORE_DIFFUSERS_VERSION=0.38.0
TRANSFORMERS_VERSION=5.11.0
UV_VERSION=0.11.26
DEEPDIFF_VERSION=9.1.0
GGUF_VERSION=0.19.0
TOMLKIT_VERSION=0.15.0
PEFT_VERSION=0.19.1
ACCELERATE_VERSION=1.14.0
HF_HUB_VERSION=1.19.0
HF_TRANSFER_VERSION=0.1.9
# ========================================
# VERSION PINS (CONTROL/RUNTIME WEB STACK)
# ========================================
FASTAPI_VERSION=0.139.0
UVICORN_VERSION=0.50.0
PYDANTIC_VERSION=2.13.4
PYTHON_MULTIPART_VERSION=0.0.32
FLASK_VERSION=3.1.3
FLASK_CORS_VERSION=6.0.5
REQUESTS_VERSION=2.34.2
PYTHON_DOTENV_VERSION=1.2.2
PYTHON_SOCKETIO_VERSION=5.16.3
WEBSOCKETS_VERSION=16.0
HTTPX_VERSION=0.28.1
# ========================================
# VERSION PINS (INVOKE VENV, USES INVOKE-SUPPORTED CUDA TORCH STACK)
# ========================================
INVOKEAI_VERSION=6.13.5
INVOKE_TORCH_VERSION=2.7.1+cu128
INVOKE_TORCHVISION_VERSION=0.22.1+cu128
INVOKE_TORCH_INDEX_URL=https://download.pytorch.org/whl/cu128
INVOKE_XFORMERS_VERSION=0.0.31.post1
INVOKE_DIFFUSERS_VERSION=0.37.0
INVOKE_TRANSFORMERS_VERSION=5.5.4
INVOKE_ACCELERATE_VERSION=1.14.0
INVOKE_HF_HUB_VERSION=1.22.0
# ========================================
# SYSTEM PACKAGES / TOOLS
# ========================================
CUDA_NVCC_PKG=cuda-nvcc-13-0
CROC_VERSION=10.4.2