truss deployment example
Browse files- truss/config.yaml +9 -11
truss/config.yaml
CHANGED
|
@@ -1,4 +1,7 @@
|
|
|
|
|
|
|
|
| 1 |
model_name: glm-5.2-vision
|
|
|
|
| 2 |
model_metadata:
|
| 3 |
example_model_input:
|
| 4 |
model: glm-5.2-vision
|
|
@@ -18,13 +21,12 @@ model_metadata:
|
|
| 18 |
|
| 19 |
# Deploy GLM-5.2-Vision on Baseten with Truss.
|
| 20 |
#
|
| 21 |
-
#
|
| 22 |
-
# assembly here: the model_cache repo below is already the fully assembled
|
| 23 |
-
# checkpoint (GLM-5.2 text + MoonViT vision tower + trained projector + processor
|
| 24 |
-
# remote code), so start_server.sh only installs the plugin and launches SGLang.
|
| 25 |
#
|
| 26 |
-
#
|
| 27 |
-
#
|
|
|
|
|
|
|
| 28 |
base_image:
|
| 29 |
# SGLang 0.5.13, CUDA 13, Blackwell (sm_100). Ships the DSA sparse-attention and
|
| 30 |
# trtllm-gen fp4-MoE kernels this checkpoint needs.
|
|
@@ -39,8 +41,7 @@ docker_server:
|
|
| 39 |
server_port: 8000
|
| 40 |
|
| 41 |
model_cache:
|
| 42 |
-
#
|
| 43 |
-
# Blackwell-only NVFP4 build, push config_nvfp4.yaml instead.
|
| 44 |
- repo_id: baseten/GLM-5.2-Vision-FP8
|
| 45 |
revision: main
|
| 46 |
volume_folder: glm5v
|
|
@@ -50,9 +51,6 @@ environment_variables:
|
|
| 50 |
# Must match volume_folder above.
|
| 51 |
GLM5V_CKPT: /app/model_cache/glm5v
|
| 52 |
|
| 53 |
-
secrets:
|
| 54 |
-
hf_access_token: null
|
| 55 |
-
|
| 56 |
resources:
|
| 57 |
accelerator: B200:8
|
| 58 |
use_gpu: true
|
|
|
|
| 1 |
+
# yaml-language-server: $schema=https://raw.githubusercontent.com/basetenlabs/truss/main/truss/config.schema.json
|
| 2 |
+
|
| 3 |
model_name: glm-5.2-vision
|
| 4 |
+
description: GLM-5.2-Vision FP8 with 1M context on 8 B200 GPUs.
|
| 5 |
model_metadata:
|
| 6 |
example_model_input:
|
| 7 |
model: glm-5.2-vision
|
|
|
|
| 21 |
|
| 22 |
# Deploy GLM-5.2-Vision on Baseten with Truss.
|
| 23 |
#
|
| 24 |
+
# uvx truss push --config config.yaml
|
|
|
|
|
|
|
|
|
|
| 25 |
#
|
| 26 |
+
# There is no checkpoint assembly here: the public Hugging Face repo below is
|
| 27 |
+
# already the fully assembled checkpoint (GLM-5.2 text + MoonViT vision tower +
|
| 28 |
+
# trained projector + processor remote code), so start_server.sh only installs
|
| 29 |
+
# the plugin and launches SGLang.
|
| 30 |
base_image:
|
| 31 |
# SGLang 0.5.13, CUDA 13, Blackwell (sm_100). Ships the DSA sparse-attention and
|
| 32 |
# trtllm-gen fp4-MoE kernels this checkpoint needs.
|
|
|
|
| 41 |
server_port: 8000
|
| 42 |
|
| 43 |
model_cache:
|
| 44 |
+
# Public Hugging Face repo: no Hugging Face token or Baseten secret is needed.
|
|
|
|
| 45 |
- repo_id: baseten/GLM-5.2-Vision-FP8
|
| 46 |
revision: main
|
| 47 |
volume_folder: glm5v
|
|
|
|
| 51 |
# Must match volume_folder above.
|
| 52 |
GLM5V_CKPT: /app/model_cache/glm5v
|
| 53 |
|
|
|
|
|
|
|
|
|
|
| 54 |
resources:
|
| 55 |
accelerator: B200:8
|
| 56 |
use_gpu: true
|