samim2024 commited on
Commit
39c1423
·
verified ·
1 Parent(s): 7dbc787

Training in progress, epoch 1

Browse files
README.md CHANGED
@@ -1,17 +1,17 @@
1
  ---
2
- base_model: facebook/opt-350m
3
  library_name: transformers
4
  model_name: tiny-chatbot-model-dpo
5
  tags:
6
  - generated_from_trainer
7
- - dpo
8
  - trl
 
9
  licence: license
10
  ---
11
 
12
  # Model Card for tiny-chatbot-model-dpo
13
 
14
- This model is a fine-tuned version of [facebook/opt-350m](https://huggingface.co/facebook/opt-350m).
15
  It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
  ## Quick start
@@ -35,8 +35,8 @@ This model was trained with DPO, a method introduced in [Direct Preference Optim
35
  ### Framework versions
36
 
37
  - TRL: 0.21.0
38
- - Transformers: 4.55.1
39
- - Pytorch: 2.6.0+cu124
40
  - Datasets: 4.0.0
41
  - Tokenizers: 0.21.4
42
 
 
1
  ---
2
+ base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
3
  library_name: transformers
4
  model_name: tiny-chatbot-model-dpo
5
  tags:
6
  - generated_from_trainer
 
7
  - trl
8
+ - dpo
9
  licence: license
10
  ---
11
 
12
  # Model Card for tiny-chatbot-model-dpo
13
 
14
+ This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-Chat-v1.0](https://huggingface.co/TinyLlama/TinyLlama-1.1B-Chat-v1.0).
15
  It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
  ## Quick start
 
35
  ### Framework versions
36
 
37
  - TRL: 0.21.0
38
+ - Transformers: 4.55.2
39
+ - Pytorch: 2.8.0+cu126
40
  - Datasets: 4.0.0
41
  - Tokenizers: 0.21.4
42
 
adapter_config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "alpha_pattern": {},
3
  "auto_mapping": null,
4
- "base_model_name_or_path": "facebook/opt-350m",
5
  "bias": "none",
6
  "corda_config": null,
7
  "eva_config": null,
 
1
  {
2
  "alpha_pattern": {},
3
  "auto_mapping": null,
4
+ "base_model_name_or_path": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
5
  "bias": "none",
6
  "corda_config": null,
7
  "eva_config": null,
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9187514ef9e43e27875c3731c26807f988463dd8c9f8b5910e91d346ece7ee90
3
- size 3159096
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0710dbbdb085ae0209bd3f83e2345eaee093665dbe11a297b21432f969dde850
3
+ size 4517152
chat_template.jinja ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {% for message in messages %}
2
+ {% if message['role'] == 'user' %}
3
+ {{ '<|user|>
4
+ ' + message['content'] + eos_token }}
5
+ {% elif message['role'] == 'system' %}
6
+ {{ '<|system|>
7
+ ' + message['content'] + eos_token }}
8
+ {% elif message['role'] == 'assistant' %}
9
+ {{ '<|assistant|>
10
+ ' + message['content'] + eos_token }}
11
+ {% endif %}
12
+ {% if loop.last and add_generation_prompt %}
13
+ {{ '<|assistant|>' }}
14
+ {% endif %}
15
+ {% endfor %}
special_tokens_map.json CHANGED
@@ -1,29 +1,29 @@
1
  {
2
  "bos_token": {
3
- "content": "</s>",
4
  "lstrip": false,
5
- "normalized": true,
6
  "rstrip": false,
7
  "single_word": false
8
  },
9
  "eos_token": {
10
  "content": "</s>",
11
  "lstrip": false,
12
- "normalized": true,
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
  "pad_token": {
17
- "content": "<pad>",
18
  "lstrip": false,
19
- "normalized": true,
20
  "rstrip": false,
21
  "single_word": false
22
  },
23
  "unk_token": {
24
- "content": "</s>",
25
  "lstrip": false,
26
- "normalized": true,
27
  "rstrip": false,
28
  "single_word": false
29
  }
 
1
  {
2
  "bos_token": {
3
+ "content": "<s>",
4
  "lstrip": false,
5
+ "normalized": false,
6
  "rstrip": false,
7
  "single_word": false
8
  },
9
  "eos_token": {
10
  "content": "</s>",
11
  "lstrip": false,
12
+ "normalized": false,
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
  "pad_token": {
17
+ "content": "</s>",
18
  "lstrip": false,
19
+ "normalized": false,
20
  "rstrip": false,
21
  "single_word": false
22
  },
23
  "unk_token": {
24
+ "content": "<unk>",
25
  "lstrip": false,
26
+ "normalized": false,
27
  "rstrip": false,
28
  "single_word": false
29
  }
tokenizer.json CHANGED
The diff for this file is too large to render. See raw diff
 
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
tokenizer_config.json CHANGED
@@ -1,11 +1,20 @@
1
  {
2
  "add_bos_token": true,
3
- "add_prefix_space": false,
 
4
  "added_tokens_decoder": {
 
 
 
 
 
 
 
 
5
  "1": {
6
- "content": "<pad>",
7
  "lstrip": false,
8
- "normalized": true,
9
  "rstrip": false,
10
  "single_word": false,
11
  "special": true
@@ -13,19 +22,22 @@
13
  "2": {
14
  "content": "</s>",
15
  "lstrip": false,
16
- "normalized": true,
17
  "rstrip": false,
18
  "single_word": false,
19
  "special": true
20
  }
21
  },
22
- "bos_token": "</s>",
23
  "clean_up_tokenization_spaces": false,
24
  "eos_token": "</s>",
25
- "errors": "replace",
26
  "extra_special_tokens": {},
27
- "model_max_length": 1000000000000000019884624838656,
28
- "pad_token": "<pad>",
29
- "tokenizer_class": "GPT2Tokenizer",
30
- "unk_token": "</s>"
 
 
 
 
31
  }
 
1
  {
2
  "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
  "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<unk>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
  "1": {
15
+ "content": "<s>",
16
  "lstrip": false,
17
+ "normalized": false,
18
  "rstrip": false,
19
  "single_word": false,
20
  "special": true
 
22
  "2": {
23
  "content": "</s>",
24
  "lstrip": false,
25
+ "normalized": false,
26
  "rstrip": false,
27
  "single_word": false,
28
  "special": true
29
  }
30
  },
31
+ "bos_token": "<s>",
32
  "clean_up_tokenization_spaces": false,
33
  "eos_token": "</s>",
 
34
  "extra_special_tokens": {},
35
+ "legacy": false,
36
+ "model_max_length": 2048,
37
+ "pad_token": "</s>",
38
+ "padding_side": "left",
39
+ "sp_model_kwargs": {},
40
+ "tokenizer_class": "LlamaTokenizer",
41
+ "unk_token": "<unk>",
42
+ "use_default_system_prompt": false
43
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:04f648754ec3caf63a70a74ee59c5e441363b9fe880a4c77d48189d144dc1d9c
3
- size 6392
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d16d997a3daec92cfcc123c87163a83edfec0be4b8a2e957d8625ddf3ccb2ce2
3
+ size 6737