File size: 236 Bytes
17b9330
 
 
 
 
 
 
1
2
3
4
5
6
7
8
{
  "model_type": "gpt2",  // Make sure this matches your model
  "architectures": ["GPT2LMHeadModel"],  // Specifies the architecture for text generation
  "tokenizer_class": "GPT2Tokenizer",
  "max_length": 100,
  "use_cache": true
}