Text Generation
Transformers
Safetensors
llama
conversational
Inference Endpoints
text-generation-inference
jondurbin commited on
Commit
05b82f6
1 Parent(s): e1f4963

Upload LlamaForCausalLM

Browse files
Files changed (2) hide show
  1. config.json +8 -4
  2. generation_config.json +1 -1
config.json CHANGED
@@ -1,8 +1,10 @@
1
  {
2
- "_name_or_path": "bagel-7b-v0.1-workdir/checkpoint-11800",
3
  "architectures": [
4
- "MistralForCausalLM"
5
  ],
 
 
6
  "bos_token_id": 1,
7
  "eos_token_id": 2,
8
  "hidden_act": "silu",
@@ -10,16 +12,18 @@
10
  "initializer_range": 0.02,
11
  "intermediate_size": 14336,
12
  "max_position_embeddings": 32768,
13
- "model_type": "mistral",
14
  "num_attention_heads": 32,
15
  "num_hidden_layers": 32,
16
  "num_key_value_heads": 8,
 
17
  "rms_norm_eps": 1e-05,
 
18
  "rope_theta": 10000.0,
19
  "sliding_window": 4096,
20
  "tie_word_embeddings": false,
21
  "torch_dtype": "bfloat16",
22
- "transformers_version": "4.35.2",
23
  "use_cache": false,
24
  "vocab_size": 32000
25
  }
 
1
  {
2
+ "_name_or_path": "bagel-dpo-7b-v0.1",
3
  "architectures": [
4
+ "LlamaForCausalLM"
5
  ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
  "bos_token_id": 1,
9
  "eos_token_id": 2,
10
  "hidden_act": "silu",
 
12
  "initializer_range": 0.02,
13
  "intermediate_size": 14336,
14
  "max_position_embeddings": 32768,
15
+ "model_type": "llama",
16
  "num_attention_heads": 32,
17
  "num_hidden_layers": 32,
18
  "num_key_value_heads": 8,
19
+ "pretraining_tp": 1,
20
  "rms_norm_eps": 1e-05,
21
+ "rope_scaling": null,
22
  "rope_theta": 10000.0,
23
  "sliding_window": 4096,
24
  "tie_word_embeddings": false,
25
  "torch_dtype": "bfloat16",
26
+ "transformers_version": "4.36.2",
27
  "use_cache": false,
28
  "vocab_size": 32000
29
  }
generation_config.json CHANGED
@@ -2,6 +2,6 @@
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
  "eos_token_id": 2,
5
- "transformers_version": "4.35.2",
6
  "use_cache": false
7
  }
 
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
  "eos_token_id": 2,
5
+ "transformers_version": "4.36.2",
6
  "use_cache": false
7
  }