rinna
/

japanese-gpt-neox-small

Text Generation

text-generation-inference

Model card Files Files and versions Community

japanese-gpt-neox-small / config.json

tianyuz's picture

* 2023/03/20 Update the model weight and config files such that it can be loaded via Huggingface's official GPT-NeoX implementation.

d93a6e4 about 1 year ago

raw history blame contribute delete

No virus

569 Bytes

	{
	"architectures": [
	"GPTNeoXForCausalLM"
	],
	"bos_token_id": 2,
	"eos_token_id": 3,
	"hidden_act": "gelu",
	"hidden_size": 768,
	"initializer_range": 0.02,
	"intermediate_size": 3072,
	"layer_norm_eps": 1e-05,
	"max_position_embeddings": 2048,
	"model_type": "gpt_neox",
	"num_attention_heads": 12,
	"num_hidden_layers": 12,
	"rotary_emb_base": 10000,
	"rotary_pct": 1.0,
	"tie_word_embeddings": false,
	"tokenizer_class": "T5Tokenizer",
	"torch_dtype": "float32",
	"use_cache": true,
	"use_parallel_residual": false,
	"vocab_size": 44416
	}