edge results: init_profile variant YAMLs; recipe GPU memory figure
Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
This commit is contained in:
1 parent
66319ed159
commit
15bb6133a3
3 files changed
+200
-1
No files matched your search
@@ -0,0 +1,100 @@
|
||||
async_chunk: true
|
||||
connectors:
|
||||
connector_of_shared_memory:
|
||||
name: SharedMemoryConnector
|
||||
extra:
|
||||
codec_streaming: true
|
||||
connector_get_sleep_s: 0.01
|
||||
connector_get_max_wait_first_chunk: 3000
|
||||
connector_get_max_wait: 300
|
||||
codec_chunk_frames: 25
|
||||
codec_left_context_frames: 72
|
||||
initial_codec_chunk_frames: 1
|
||||
decode_batch_max_size: 4
|
||||
stages:
|
||||
- stage_id: 0
|
||||
max_num_seqs: 64
|
||||
gpu_memory_utilization: 0.3
|
||||
trust_remote_code: true
|
||||
enable_prefix_caching: false
|
||||
async_scheduling: true
|
||||
max_num_batched_tokens: 32768
|
||||
max_model_len: 4096
|
||||
devices: '0'
|
||||
output_connectors:
|
||||
to_stage_1: connector_of_shared_memory
|
||||
default_sampling_params:
|
||||
temperature: 0.9
|
||||
top_k: 50
|
||||
max_tokens: 4096
|
||||
min_tokens: 2
|
||||
repetition_penalty: 1.05
|
||||
silence_ban_frames: 0
|
||||
subtalker_sampling_params:
|
||||
do_sample: true
|
||||
temperature: 0.9
|
||||
top_k: 50
|
||||
top_p: 1.0
|
||||
enforce_eager: true
|
||||
- stage_id: 1
|
||||
max_num_seqs: 64
|
||||
gpu_memory_utilization: 0.3
|
||||
enforce_eager: true
|
||||
trust_remote_code: true
|
||||
enable_prefix_caching: false
|
||||
async_scheduling: true
|
||||
max_num_batched_tokens: 65536
|
||||
max_model_len: 65536
|
||||
devices: '0'
|
||||
input_connectors:
|
||||
from_stage_0: connector_of_shared_memory
|
||||
default_sampling_params:
|
||||
temperature: 0.0
|
||||
top_p: 1.0
|
||||
top_k: -1
|
||||
max_tokens: 65536
|
||||
repetition_penalty: 1.0
|
||||
platforms:
|
||||
rocm:
|
||||
stages:
|
||||
- stage_id: 1
|
||||
enforce_eager: true
|
||||
npu:
|
||||
stages:
|
||||
- stage_id: 0
|
||||
enforce_eager: false
|
||||
compilation_config:
|
||||
cudagraph_mode: PIECEWISE
|
||||
- stage_id: 1
|
||||
compilation_config:
|
||||
cudagraph_mode: PIECEWISE
|
||||
cpu:
|
||||
stages:
|
||||
- stage_id: 0
|
||||
engine_args:
|
||||
enforce_eager: true
|
||||
async_scheduling: false
|
||||
max_num_seqs: 2
|
||||
max_num_batched_tokens: 2048
|
||||
max_model_len: 2048
|
||||
gpu_memory_utilization: 0.05
|
||||
dtype: bfloat16
|
||||
kv_cache_memory_bytes: 4294967296
|
||||
runtime:
|
||||
devices: cpu
|
||||
env:
|
||||
VLLM_CPU_OMP_THREADS_BIND: 0-11
|
||||
- stage_id: 1
|
||||
engine_args:
|
||||
enforce_eager: true
|
||||
async_scheduling: false
|
||||
max_num_seqs: 2
|
||||
max_num_batched_tokens: 8192
|
||||
max_model_len: 8192
|
||||
gpu_memory_utilization: 0.05
|
||||
dtype: bfloat16
|
||||
kv_cache_memory_bytes: 2147483648
|
||||
runtime:
|
||||
devices: cpu
|
||||
env:
|
||||
VLLM_CPU_OMP_THREADS_BIND: 12-15
|
||||
+99
@@ -0,0 +1,99 @@
|
||||
async_chunk: true
|
||||
connectors:
|
||||
connector_of_shared_memory:
|
||||
name: SharedMemoryConnector
|
||||
extra:
|
||||
codec_streaming: true
|
||||
connector_get_sleep_s: 0.01
|
||||
connector_get_max_wait_first_chunk: 3000
|
||||
connector_get_max_wait: 300
|
||||
codec_chunk_frames: 25
|
||||
codec_left_context_frames: 72
|
||||
initial_codec_chunk_frames: 1
|
||||
decode_batch_max_size: 4
|
||||
stages:
|
||||
- stage_id: 0
|
||||
max_num_seqs: 64
|
||||
gpu_memory_utilization: 0.3
|
||||
trust_remote_code: true
|
||||
enable_prefix_caching: false
|
||||
async_scheduling: true
|
||||
max_num_batched_tokens: 32768
|
||||
max_model_len: 4096
|
||||
devices: '0'
|
||||
output_connectors:
|
||||
to_stage_1: connector_of_shared_memory
|
||||
default_sampling_params:
|
||||
temperature: 0.9
|
||||
top_k: 50
|
||||
max_tokens: 4096
|
||||
min_tokens: 2
|
||||
repetition_penalty: 1.05
|
||||
silence_ban_frames: 0
|
||||
subtalker_sampling_params:
|
||||
do_sample: true
|
||||
temperature: 0.9
|
||||
top_k: 50
|
||||
top_p: 1.0
|
||||
- stage_id: 1
|
||||
max_num_seqs: 64
|
||||
gpu_memory_utilization: 0.3
|
||||
enforce_eager: true
|
||||
trust_remote_code: true
|
||||
enable_prefix_caching: false
|
||||
async_scheduling: true
|
||||
max_num_batched_tokens: 65536
|
||||
max_model_len: 65536
|
||||
devices: '0'
|
||||
input_connectors:
|
||||
from_stage_0: connector_of_shared_memory
|
||||
default_sampling_params:
|
||||
temperature: 0.0
|
||||
top_p: 1.0
|
||||
top_k: -1
|
||||
max_tokens: 65536
|
||||
repetition_penalty: 1.0
|
||||
platforms:
|
||||
rocm:
|
||||
stages:
|
||||
- stage_id: 1
|
||||
enforce_eager: true
|
||||
npu:
|
||||
stages:
|
||||
- stage_id: 0
|
||||
enforce_eager: false
|
||||
compilation_config:
|
||||
cudagraph_mode: PIECEWISE
|
||||
- stage_id: 1
|
||||
compilation_config:
|
||||
cudagraph_mode: PIECEWISE
|
||||
cpu:
|
||||
stages:
|
||||
- stage_id: 0
|
||||
engine_args:
|
||||
enforce_eager: true
|
||||
async_scheduling: false
|
||||
max_num_seqs: 2
|
||||
max_num_batched_tokens: 2048
|
||||
max_model_len: 2048
|
||||
gpu_memory_utilization: 0.05
|
||||
dtype: bfloat16
|
||||
kv_cache_memory_bytes: 4294967296
|
||||
runtime:
|
||||
devices: cpu
|
||||
env:
|
||||
VLLM_CPU_OMP_THREADS_BIND: 0-11
|
||||
- stage_id: 1
|
||||
engine_args:
|
||||
enforce_eager: true
|
||||
async_scheduling: false
|
||||
max_num_seqs: 2
|
||||
max_num_batched_tokens: 8192
|
||||
max_model_len: 8192
|
||||
gpu_memory_utilization: 0.05
|
||||
dtype: bfloat16
|
||||
kv_cache_memory_bytes: 2147483648
|
||||
runtime:
|
||||
devices: cpu
|
||||
env:
|
||||
VLLM_CPU_OMP_THREADS_BIND: 12-15
|
||||
@@ -3,7 +3,7 @@
|
||||
This recipe covers the edge deploy profile, the CPU platform, hardware
|
||||
auto-adaptation, codec-token streaming with on-device decoding, and the
|
||||
profiling tools added on the `edge/omni-edge` branch. Numbers quoted here were
|
||||
measured on one L20X (48 GB) and a 2x56-core Xeon 8480C host on 2026-09-09; the
|
||||
measured on one NVIDIA L20X (143 GB) and a 2x56-core Xeon 8480C host on 2026-09-09; the
|
||||
JSON files are committed under `benchmarks/tts/edge_results/`
|
||||
(`python benchmarks/tts/edge_results/summarize.py` renders them).
|
||||
|
||||
|
||||
Reference in new issue
Block a user