config.json

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