AlexBeneath commited on
Commit
17b9330
·
verified ·
1 Parent(s): 6ce32fe

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -0
config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "gpt2", // Make sure this matches your model
3
+ "architectures": ["GPT2LMHeadModel"], // Specifies the architecture for text generation
4
+ "tokenizer_class": "GPT2Tokenizer",
5
+ "max_length": 100,
6
+ "use_cache": true
7
+ }