lvwerra HF staff chargoddard commited on
Commit
54da767
1 Parent(s): 444cd6f

Add architectures field to config.json (#2)

Browse files

- Add architectures field to config.json (df35d7804fd0ec8d5d62b9a31a9054037333c558)


Co-authored-by: Charles Goddard <[email protected]>

Files changed (1) hide show
  1. config.json +3 -0
config.json CHANGED
@@ -1,4 +1,7 @@
1
  {
 
 
 
2
  "attention_dropout": 0.0,
3
  "bos_token_id": 0,
4
  "eos_token_id": 0,
 
1
  {
2
+ "architectures": [
3
+ "Starcoder2ForCausalLM"
4
+ ],
5
  "attention_dropout": 0.0,
6
  "bos_token_id": 0,
7
  "eos_token_id": 0,