kalen22 commited on
Commit
e93eb3b
·
verified ·
1 Parent(s): b95e546

Upload model

Browse files
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: bigcode/starcoder2-3b
3
+ library_name: transformers
4
+ model_name: silver_model
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - sft
9
+ licence: license
10
+ ---
11
+
12
+ # Model Card for silver_model
13
+
14
+ This model is a fine-tuned version of [bigcode/starcoder2-3b](https://huggingface.co/bigcode/starcoder2-3b).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
+
17
+ ## Quick start
18
+
19
+ ```python
20
+ from transformers import pipeline
21
+
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="kalen22/silver_model", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
+
28
+ ## Training procedure
29
+
30
+
31
+
32
+
33
+
34
+ This model was trained with SFT.
35
+
36
+ ### Framework versions
37
+
38
+ - TRL: 0.29.1
39
+ - Transformers: 5.5.0.dev0
40
+ - Pytorch: 2.11.0
41
+ - Datasets: 4.8.4
42
+ - Tokenizers: 0.22.2
43
+
44
+ ## Citations
45
+
46
+
47
+
48
+ Cite TRL as:
49
+
50
+ ```bibtex
51
+ @software{vonwerra2020trl,
52
+ title = {{TRL: Transformers Reinforcement Learning}},
53
+ author = {von Werra, Leandro and Belkada, Younes and Tunstall, Lewis and Beeching, Edward and Thrush, Tristan and Lambert, Nathan and Huang, Shengyi and Rasul, Kashif and Gallouédec, Quentin},
54
+ license = {Apache-2.0},
55
+ url = {https://github.com/huggingface/trl},
56
+ year = {2020}
57
+ }
58
+ ```
adapter_config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "bigcode/starcoder2-3b",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "loftq_config": {},
12
+ "lora_alpha": 8,
13
+ "lora_dropout": 0.0,
14
+ "megatron_config": null,
15
+ "megatron_core": "megatron.core",
16
+ "modules_to_save": null,
17
+ "peft_type": "LORA",
18
+ "r": 8,
19
+ "rank_pattern": {},
20
+ "revision": null,
21
+ "target_modules": [
22
+ "up_proj",
23
+ "k_proj",
24
+ "o_proj",
25
+ "q_proj",
26
+ "v_proj",
27
+ "gate_proj",
28
+ "down_proj"
29
+ ],
30
+ "task_type": "CAUSAL_LM",
31
+ "use_rslora": false
32
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a3ae5f508451d363dcbfd48e7e4a7693a782441d206aa19211b5362e849dbde
3
+ size 9124880
final_checkpoint/config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Starcoder2ForCausalLM"
4
+ ],
5
+ "attention_dropout": 0.1,
6
+ "bos_token_id": 0,
7
+ "dtype": "float32",
8
+ "embedding_dropout": 0.1,
9
+ "eos_token_id": 0,
10
+ "hidden_act": "gelu_pytorch_tanh",
11
+ "hidden_size": 3072,
12
+ "initializer_range": 0.018042,
13
+ "intermediate_size": 12288,
14
+ "max_position_embeddings": 16384,
15
+ "mlp_type": "default",
16
+ "model_type": "starcoder2",
17
+ "norm_epsilon": 1e-05,
18
+ "norm_type": "layer_norm",
19
+ "num_attention_heads": 24,
20
+ "num_hidden_layers": 30,
21
+ "num_key_value_heads": 2,
22
+ "pad_token_id": null,
23
+ "quantization_config": {
24
+ "_load_in_4bit": true,
25
+ "_load_in_8bit": false,
26
+ "bnb_4bit_compute_dtype": "bfloat16",
27
+ "bnb_4bit_quant_storage": "uint8",
28
+ "bnb_4bit_quant_type": "nf4",
29
+ "bnb_4bit_use_double_quant": false,
30
+ "llm_int8_enable_fp32_cpu_offload": false,
31
+ "llm_int8_has_fp16_weight": false,
32
+ "llm_int8_skip_modules": null,
33
+ "llm_int8_threshold": 6.0,
34
+ "load_in_4bit": true,
35
+ "load_in_8bit": false,
36
+ "quant_method": "bitsandbytes"
37
+ },
38
+ "residual_dropout": 0.1,
39
+ "rope_parameters": {
40
+ "rope_theta": 999999.4420358813,
41
+ "rope_type": "default"
42
+ },
43
+ "sliding_window": 4096,
44
+ "tie_word_embeddings": true,
45
+ "transformers_version": "5.5.0.dev0",
46
+ "use_bias": true,
47
+ "use_cache": false,
48
+ "vocab_size": 49152
49
+ }
final_checkpoint/generation_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": [
5
+ 0
6
+ ],
7
+ "transformers_version": "5.5.0.dev0",
8
+ "use_cache": true
9
+ }
final_checkpoint/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76207b5d69f59c9993e0328570af22cb47cbc5becba4da6436944e92d9a96cb7
3
+ size 2236279004
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "clean_up_tokenization_spaces": true,
6
+ "eos_token": "<|endoftext|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<|endoftext|>",
10
+ "<fim_prefix>",
11
+ "<fim_middle>",
12
+ "<fim_suffix>",
13
+ "<fim_pad>",
14
+ "<repo_name>",
15
+ "<file_sep>",
16
+ "<issue_start>",
17
+ "<issue_comment>",
18
+ "<issue_closed>",
19
+ "<jupyter_start>",
20
+ "<jupyter_text>",
21
+ "<jupyter_code>",
22
+ "<jupyter_output>",
23
+ "<jupyter_script>",
24
+ "<empty_output>",
25
+ "<code_to_intermediate>",
26
+ "<intermediate_to_code>",
27
+ "<pr>",
28
+ "<pr_status>",
29
+ "<pr_is_merged>",
30
+ "<pr_base>",
31
+ "<pr_file>",
32
+ "<pr_base_code>",
33
+ "<pr_diff>",
34
+ "<pr_diff_hunk>",
35
+ "<pr_comment>",
36
+ "<pr_event_id>",
37
+ "<pr_review>",
38
+ "<pr_review_state>",
39
+ "<pr_review_comment>",
40
+ "<pr_in_reply_to_review_id>",
41
+ "<pr_in_reply_to_comment_id>",
42
+ "<pr_diff_hunk_comment_line>",
43
+ "<NAME>",
44
+ "<EMAIL>",
45
+ "<KEY>",
46
+ "<PASSWORD>"
47
+ ],
48
+ "is_local": false,
49
+ "model_max_length": 1000000000000000019884624838656,
50
+ "pad_token": null,
51
+ "tokenizer_class": "GPT2Tokenizer",
52
+ "unk_token": "<|endoftext|>",
53
+ "vocab_size": 49152
54
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a40a33bc7124d4c7cc3831d659f66638a8062b0cdeafbf7080c0a397ca821bcc
3
+ size 5649