End of training
Browse files- README.md +59 -0
- adapter_config.json +28 -0
- adapter_model.safetensors +3 -0
- runs/Mar13_05-08-36_4d8f5d5c7201/events.out.tfevents.1710306517.4d8f5d5c7201.33.0 +3 -0
- special_tokens_map.json +35 -0
- tokenizer.model +3 -0
- tokenizer_config.json +73 -0
- training_args.bin +3 -0
    	
        README.md
    ADDED
    
    | @@ -0,0 +1,59 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            ---
         | 
| 2 | 
            +
            license: other
         | 
| 3 | 
            +
            library_name: peft
         | 
| 4 | 
            +
            tags:
         | 
| 5 | 
            +
            - generated_from_trainer
         | 
| 6 | 
            +
            base_model: 01-ai/Yi-6B-Chat
         | 
| 7 | 
            +
            model-index:
         | 
| 8 | 
            +
            - name: Mental-Yi-7B-V1
         | 
| 9 | 
            +
              results: []
         | 
| 10 | 
            +
            ---
         | 
| 11 | 
            +
             | 
| 12 | 
            +
            <!-- This model card has been generated automatically according to the information the Trainer had access to. You
         | 
| 13 | 
            +
            should probably proofread and complete it, then remove this comment. -->
         | 
| 14 | 
            +
             | 
| 15 | 
            +
            # Mental-Yi-7B-V1
         | 
| 16 | 
            +
             | 
| 17 | 
            +
            This model is a fine-tuned version of [01-ai/Yi-6B-Chat](https://huggingface.co/01-ai/Yi-6B-Chat) on an unknown dataset.
         | 
| 18 | 
            +
            It achieves the following results on the evaluation set:
         | 
| 19 | 
            +
            - eval_loss: 2.0950
         | 
| 20 | 
            +
            - eval_runtime: 229.6164
         | 
| 21 | 
            +
            - eval_samples_per_second: 4.172
         | 
| 22 | 
            +
            - eval_steps_per_second: 0.523
         | 
| 23 | 
            +
            - epoch: 8.8
         | 
| 24 | 
            +
            - step: 8000
         | 
| 25 | 
            +
             | 
| 26 | 
            +
            ## Model description
         | 
| 27 | 
            +
             | 
| 28 | 
            +
            More information needed
         | 
| 29 | 
            +
             | 
| 30 | 
            +
            ## Intended uses & limitations
         | 
| 31 | 
            +
             | 
| 32 | 
            +
            More information needed
         | 
| 33 | 
            +
             | 
| 34 | 
            +
            ## Training and evaluation data
         | 
| 35 | 
            +
             | 
| 36 | 
            +
            More information needed
         | 
| 37 | 
            +
             | 
| 38 | 
            +
            ## Training procedure
         | 
| 39 | 
            +
             | 
| 40 | 
            +
            ### Training hyperparameters
         | 
| 41 | 
            +
             | 
| 42 | 
            +
            The following hyperparameters were used during training:
         | 
| 43 | 
            +
            - learning_rate: 2e-05
         | 
| 44 | 
            +
            - train_batch_size: 8
         | 
| 45 | 
            +
            - eval_batch_size: 8
         | 
| 46 | 
            +
            - seed: 42
         | 
| 47 | 
            +
            - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
         | 
| 48 | 
            +
            - lr_scheduler_type: cosine
         | 
| 49 | 
            +
            - lr_scheduler_warmup_ratio: 0.1
         | 
| 50 | 
            +
            - num_epochs: 10
         | 
| 51 | 
            +
            - mixed_precision_training: Native AMP
         | 
| 52 | 
            +
             | 
| 53 | 
            +
            ### Framework versions
         | 
| 54 | 
            +
             | 
| 55 | 
            +
            - PEFT 0.9.0
         | 
| 56 | 
            +
            - Transformers 4.38.2
         | 
| 57 | 
            +
            - Pytorch 2.1.2
         | 
| 58 | 
            +
            - Datasets 2.18.0
         | 
| 59 | 
            +
            - Tokenizers 0.15.1
         | 
    	
        adapter_config.json
    ADDED
    
    | @@ -0,0 +1,28 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "alpha_pattern": {},
         | 
| 3 | 
            +
              "auto_mapping": null,
         | 
| 4 | 
            +
              "base_model_name_or_path": "01-ai/Yi-6B-Chat",
         | 
| 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": 16,
         | 
| 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 | 
            +
                "q_proj",
         | 
| 23 | 
            +
                "v_proj"
         | 
| 24 | 
            +
              ],
         | 
| 25 | 
            +
              "task_type": "CAUSAL_LM",
         | 
| 26 | 
            +
              "use_dora": false,
         | 
| 27 | 
            +
              "use_rslora": false
         | 
| 28 | 
            +
            }
         | 
    	
        adapter_model.safetensors
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:ea4e0ec342ad1046a1570aa4e95d516b634fa1f974c314ef584aaa48137df4cc
         | 
| 3 | 
            +
            size 13124104
         | 
    	
        runs/Mar13_05-08-36_4d8f5d5c7201/events.out.tfevents.1710306517.4d8f5d5c7201.33.0
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:fc63fbcb7ddbe0dafcaf7663db33d89896cc5c3fdef90dba735a83474daca5df
         | 
| 3 | 
            +
            size 8890
         | 
    	
        special_tokens_map.json
    ADDED
    
    | @@ -0,0 +1,35 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "additional_special_tokens": [
         | 
| 3 | 
            +
                "<|im_start|>",
         | 
| 4 | 
            +
                "<|im_end|>",
         | 
| 5 | 
            +
                "<|im_sep|>"
         | 
| 6 | 
            +
              ],
         | 
| 7 | 
            +
              "bos_token": {
         | 
| 8 | 
            +
                "content": "<|startoftext|>",
         | 
| 9 | 
            +
                "lstrip": false,
         | 
| 10 | 
            +
                "normalized": true,
         | 
| 11 | 
            +
                "rstrip": false,
         | 
| 12 | 
            +
                "single_word": false
         | 
| 13 | 
            +
              },
         | 
| 14 | 
            +
              "eos_token": {
         | 
| 15 | 
            +
                "content": "<|endoftext|>",
         | 
| 16 | 
            +
                "lstrip": false,
         | 
| 17 | 
            +
                "normalized": true,
         | 
| 18 | 
            +
                "rstrip": false,
         | 
| 19 | 
            +
                "single_word": false
         | 
| 20 | 
            +
              },
         | 
| 21 | 
            +
              "pad_token": {
         | 
| 22 | 
            +
                "content": "<unk>",
         | 
| 23 | 
            +
                "lstrip": false,
         | 
| 24 | 
            +
                "normalized": true,
         | 
| 25 | 
            +
                "rstrip": false,
         | 
| 26 | 
            +
                "single_word": false
         | 
| 27 | 
            +
              },
         | 
| 28 | 
            +
              "unk_token": {
         | 
| 29 | 
            +
                "content": "<unk>",
         | 
| 30 | 
            +
                "lstrip": false,
         | 
| 31 | 
            +
                "normalized": true,
         | 
| 32 | 
            +
                "rstrip": false,
         | 
| 33 | 
            +
                "single_word": false
         | 
| 34 | 
            +
              }
         | 
| 35 | 
            +
            }
         | 
    	
        tokenizer.model
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:386c49cf943d71aa110361135338c50e38beeff0a66593480421f37b319e1a39
         | 
| 3 | 
            +
            size 1033105
         | 
    	
        tokenizer_config.json
    ADDED
    
    | @@ -0,0 +1,73 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "add_bos_token": false,
         | 
| 3 | 
            +
              "add_eos_token": false,
         | 
| 4 | 
            +
              "add_prefix_space": true,
         | 
| 5 | 
            +
              "added_tokens_decoder": {
         | 
| 6 | 
            +
                "0": {
         | 
| 7 | 
            +
                  "content": "<unk>",
         | 
| 8 | 
            +
                  "lstrip": false,
         | 
| 9 | 
            +
                  "normalized": true,
         | 
| 10 | 
            +
                  "rstrip": false,
         | 
| 11 | 
            +
                  "single_word": false,
         | 
| 12 | 
            +
                  "special": true
         | 
| 13 | 
            +
                },
         | 
| 14 | 
            +
                "1": {
         | 
| 15 | 
            +
                  "content": "<|startoftext|>",
         | 
| 16 | 
            +
                  "lstrip": false,
         | 
| 17 | 
            +
                  "normalized": true,
         | 
| 18 | 
            +
                  "rstrip": false,
         | 
| 19 | 
            +
                  "single_word": false,
         | 
| 20 | 
            +
                  "special": true
         | 
| 21 | 
            +
                },
         | 
| 22 | 
            +
                "2": {
         | 
| 23 | 
            +
                  "content": "<|endoftext|>",
         | 
| 24 | 
            +
                  "lstrip": false,
         | 
| 25 | 
            +
                  "normalized": true,
         | 
| 26 | 
            +
                  "rstrip": false,
         | 
| 27 | 
            +
                  "single_word": false,
         | 
| 28 | 
            +
                  "special": true
         | 
| 29 | 
            +
                },
         | 
| 30 | 
            +
                "6": {
         | 
| 31 | 
            +
                  "content": "<|im_start|>",
         | 
| 32 | 
            +
                  "lstrip": false,
         | 
| 33 | 
            +
                  "normalized": false,
         | 
| 34 | 
            +
                  "rstrip": false,
         | 
| 35 | 
            +
                  "single_word": false,
         | 
| 36 | 
            +
                  "special": true
         | 
| 37 | 
            +
                },
         | 
| 38 | 
            +
                "7": {
         | 
| 39 | 
            +
                  "content": "<|im_end|>",
         | 
| 40 | 
            +
                  "lstrip": false,
         | 
| 41 | 
            +
                  "normalized": false,
         | 
| 42 | 
            +
                  "rstrip": false,
         | 
| 43 | 
            +
                  "single_word": false,
         | 
| 44 | 
            +
                  "special": true
         | 
| 45 | 
            +
                },
         | 
| 46 | 
            +
                "8": {
         | 
| 47 | 
            +
                  "content": "<|im_sep|>",
         | 
| 48 | 
            +
                  "lstrip": false,
         | 
| 49 | 
            +
                  "normalized": false,
         | 
| 50 | 
            +
                  "rstrip": false,
         | 
| 51 | 
            +
                  "single_word": false,
         | 
| 52 | 
            +
                  "special": true
         | 
| 53 | 
            +
                }
         | 
| 54 | 
            +
              },
         | 
| 55 | 
            +
              "additional_special_tokens": [
         | 
| 56 | 
            +
                "<|im_start|>",
         | 
| 57 | 
            +
                "<|im_end|>",
         | 
| 58 | 
            +
                "<|im_sep|>"
         | 
| 59 | 
            +
              ],
         | 
| 60 | 
            +
              "bos_token": "<|startoftext|>",
         | 
| 61 | 
            +
              "chat_template": "{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
         | 
| 62 | 
            +
              "clean_up_tokenization_spaces": false,
         | 
| 63 | 
            +
              "eos_token": "<|endoftext|>",
         | 
| 64 | 
            +
              "legacy": true,
         | 
| 65 | 
            +
              "model_max_length": 4096,
         | 
| 66 | 
            +
              "pad_token": "<unk>",
         | 
| 67 | 
            +
              "padding_side": "right",
         | 
| 68 | 
            +
              "sp_model_kwargs": {},
         | 
| 69 | 
            +
              "spaces_between_special_tokens": false,
         | 
| 70 | 
            +
              "tokenizer_class": "LlamaTokenizer",
         | 
| 71 | 
            +
              "unk_token": "<unk>",
         | 
| 72 | 
            +
              "use_default_system_prompt": true
         | 
| 73 | 
            +
            }
         | 
    	
        training_args.bin
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:c2c10e3e8e93808747c24fbce121a4ae47aaa897d079547ea846cf05eef3802f
         | 
| 3 | 
            +
            size 4856
         | 
