vysri commited on
Commit
8c92a82
·
verified ·
1 Parent(s): 5893b94

Upload folder using huggingface_hub

Browse files
chat_template.jinja ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {% for message in messages %}{% if loop.first and messages[0]['role'] != 'system' %}{{ '<|im_start|>system
2
+ You are a helpful AI assistant named SmolLM, trained by Hugging Face<|im_end|>
3
+ ' }}{% endif %}{{'<|im_start|>' + message['role'] + '
4
+ ' + message['content'] + '<|im_end|>' + '
5
+ '}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant
6
+ ' }}{% endif %}
config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 1,
8
+ "dtype": "bfloat16",
9
+ "eos_token_id": 2,
10
+ "head_dim": 64,
11
+ "hidden_act": "silu",
12
+ "hidden_size": 960,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 2560,
15
+ "is_llama_config": true,
16
+ "max_position_embeddings": 8192,
17
+ "mlp_bias": false,
18
+ "model_type": "llama",
19
+ "num_attention_heads": 15,
20
+ "num_hidden_layers": 32,
21
+ "num_key_value_heads": 5,
22
+ "pad_token_id": 2,
23
+ "pretraining_tp": 1,
24
+ "rms_norm_eps": 1e-05,
25
+ "rope_interleaved": false,
26
+ "rope_parameters": {
27
+ "rope_theta": 100000,
28
+ "rope_type": "default"
29
+ },
30
+ "tie_word_embeddings": true,
31
+ "transformers.js_config": {
32
+ "kv_cache_dtype": {
33
+ "fp16": "float16",
34
+ "q4f16": "float16"
35
+ }
36
+ },
37
+ "transformers_version": "5.5.4",
38
+ "use_cache": true,
39
+ "vocab_size": 49153
40
+ }
config.snapshot.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "HuggingFaceTB/SmolLM2-360M-Instruct",
3
+ "dataset_path": "your_dataset_path_here",
4
+ "dataset_mode": "full",
5
+ "platform": "gpu",
6
+ "wanb_project": "convfill",
7
+ "training_split_percent": 0.9,
8
+ "add_special_tokens": true,
9
+ "special_tokens": {
10
+ "additional_special_tokens": [
11
+ "<sil>"
12
+ ]
13
+ },
14
+ "boundary_tokens": {
15
+ "USER_START": "<|im_start|>user\n",
16
+ "KNOWLEDGE_START": "<|im_start|>knowledge\n",
17
+ "ASSN_START": "<|im_start|>assistant\n",
18
+ "END": "<|im_end|>\n"
19
+ },
20
+ "training_config": {
21
+ "batch_size": 4,
22
+ "accumulate_grad_batches": 4,
23
+ "num_epochs": 1,
24
+ "learning_rate": 0.0001,
25
+ "weight_decay": 0.01,
26
+ "warmup_steps": 200,
27
+ "scheduler": "cosine",
28
+ "val_check_interval": 0.25
29
+ },
30
+ "logging_config": {
31
+ "log_every_n_steps": 10
32
+ },
33
+ "checkpoint_config": {
34
+ "every_n_train_steps": 50
35
+ }
36
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 2,
5
+ "pad_token_id": 2,
6
+ "transformers_version": "5.5.4"
7
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77219a917ed0e409447541d8397f02b7dc9f5919b41c43fc6c9bf8ddcb080a25
3
+ size 723676832
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|im_start|>",
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<sil>"
10
+ ],
11
+ "is_local": false,
12
+ "model_max_length": 8192,
13
+ "pad_token": "<|im_end|>",
14
+ "tokenizer_class": "GPT2Tokenizer",
15
+ "unk_token": "<|endoftext|>",
16
+ "vocab_size": 49152
17
+ }