ayjays132 commited on
Commit
67b5009
1 Parent(s): beb17a9

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -26,7 +26,7 @@
26
  "max_sequence_length": 200,
27
  "min_epochs_before_growth": 5,
28
  "model_filename": "pytorch_model.bin",
29
- "model_type": "llama",
30
  "num_embeddings": 25000,
31
  "pruning_improvement_threshold": 0.005,
32
  "some_adaptation_rate": 0.05,
 
26
  "max_sequence_length": 200,
27
  "min_epochs_before_growth": 5,
28
  "model_filename": "pytorch_model.bin",
29
+ "model_type": "GPT2LMHeadModel",
30
  "num_embeddings": 25000,
31
  "pruning_improvement_threshold": 0.005,
32
  "some_adaptation_rate": 0.05,