config.json

{
  "architectures": [
    "Classification_model"
  ],
  "device": "cuda",
  "dropout_prob": 0.1,
  "embed_dim": 128,
  "ff_dim": 512,
  "max_length": 128,
  "model_type": "transformer",
  "num_classess": 6,
  "num_heads": 8,
  "num_layers": 10,
  "pre_normalization": true,
  "torch_dtype": "float32",
  "transformers_version": "4.49.0",
  "vocab_size": 30522
}