config.json

{
  "architectures": [
    "QueryCompletionModelForHub"
  ],
  "dropout": 0.1,
  "embed_dim": 256,
  "filter_sizes": [
    3,
    4,
    5
  ],
  "model_type": "query-completion",
  "num_filters": 64,
  "num_heads": 4,
  "num_transformer_layers": 2,
  "pretrained_model_name": "google/byt5-small",
  "torch_dtype": "float32",
  "transformers_version": "4.50.3",
  "use_pretrained_embeddings": true,
  "vocab_size": 384,
  "auto_map": {
    "AutoConfig": "modeling_query_completion.QueryCompletionConfig",
    "AutoModel": "modeling_query_completion.QueryCompletionModelForHub"
  }
}