switch4gen / config.json
ZhankuiHe's picture
Upload Switch4GenModule
6c559c2 verified
{
"_name_or_path": "gpt8gen0.01",
"architectures": [
"Switch4GenModule"
],
"dialog_model": "gpt2",
"model_type": "switch4gen",
"num_hidden_layers": 4,
"stop_tokens": [
50257
],
"switch_token_id": 50257,
"torch_dtype": "float32",
"transformers_version": "4.44.0"
}