tabby/testdata/gptneox/models/fastertransformer/1/1-gpu/config.ini

13 lines
208 B
INI
Raw Normal View History

2023-03-20 08:51:28 +00:00
[gptneox]
model_name = gptneox
head_num = 12
size_per_head = 64
inter_size = 3072
num_layer = 12
rotary_embedding = 64
vocab_size = 50304
start_id = 0
end_id = 0
use_gptj_residual = 0
weight_data_type = fp32