config.json

{
  "architectures": [
    "Switch4GenModule"
  ],
  "dialog_model": "gpt2",
  "model_type": "switch4gen",
  "num_hidden_layers": 4,
  "stop_tokens": [
    50256,
    50257
  ],
  "switch_token_id": 50257,
  "torch_dtype": "float32",
  "transformers_version": "4.44.0"
}