Training in progress, epoch 1
Browse files- .gitattributes +1 -0
- README.md +58 -0
- added_tokens.json +3 -0
- chat_template.jinja +47 -0
- config.json +54 -0
- model.safetensors +3 -0
- runs/Aug14_23-55-20_b500db30f9fe/events.out.tfevents.1755215734.b500db30f9fe.400.0 +3 -0
- special_tokens_map.json +33 -0
- tokenizer.json +3 -0
- tokenizer.model +3 -0
- tokenizer_config.json +0 -0
- training_args.bin +3 -0
    	
        .gitattributes
    CHANGED
    
    | @@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text | |
| 33 | 
             
            *.zip filter=lfs diff=lfs merge=lfs -text
         | 
| 34 | 
             
            *.zst filter=lfs diff=lfs merge=lfs -text
         | 
| 35 | 
             
            *tfevents* filter=lfs diff=lfs merge=lfs -text
         | 
|  | 
|  | |
| 33 | 
             
            *.zip filter=lfs diff=lfs merge=lfs -text
         | 
| 34 | 
             
            *.zst filter=lfs diff=lfs merge=lfs -text
         | 
| 35 | 
             
            *tfevents* filter=lfs diff=lfs merge=lfs -text
         | 
| 36 | 
            +
            tokenizer.json filter=lfs diff=lfs merge=lfs -text
         | 
    	
        README.md
    ADDED
    
    | @@ -0,0 +1,58 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            ---
         | 
| 2 | 
            +
            base_model: google/gemma-3-270m-it
         | 
| 3 | 
            +
            library_name: transformers
         | 
| 4 | 
            +
            model_name: MyGemmaNPC
         | 
| 5 | 
            +
            tags:
         | 
| 6 | 
            +
            - generated_from_trainer
         | 
| 7 | 
            +
            - trl
         | 
| 8 | 
            +
            - sft
         | 
| 9 | 
            +
            licence: license
         | 
| 10 | 
            +
            ---
         | 
| 11 | 
            +
             | 
| 12 | 
            +
            # Model Card for MyGemmaNPC
         | 
| 13 | 
            +
             | 
| 14 | 
            +
            This model is a fine-tuned version of [google/gemma-3-270m-it](https://huggingface.co/google/gemma-3-270m-it).
         | 
| 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="wooii/MyGemmaNPC", 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 | 
            +
            This model was trained with SFT.
         | 
| 34 | 
            +
             | 
| 35 | 
            +
            ### Framework versions
         | 
| 36 | 
            +
             | 
| 37 | 
            +
            - TRL: 0.21.0
         | 
| 38 | 
            +
            - Transformers: 4.55.0
         | 
| 39 | 
            +
            - Pytorch: 2.6.0+cu124
         | 
| 40 | 
            +
            - Datasets: 4.0.0
         | 
| 41 | 
            +
            - Tokenizers: 0.21.4
         | 
| 42 | 
            +
             | 
| 43 | 
            +
            ## Citations
         | 
| 44 | 
            +
             | 
| 45 | 
            +
             | 
| 46 | 
            +
             | 
| 47 | 
            +
            Cite TRL as:
         | 
| 48 | 
            +
                
         | 
| 49 | 
            +
            ```bibtex
         | 
| 50 | 
            +
            @misc{vonwerra2022trl,
         | 
| 51 | 
            +
            	title        = {{TRL: Transformer Reinforcement Learning}},
         | 
| 52 | 
            +
            	author       = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
         | 
| 53 | 
            +
            	year         = 2020,
         | 
| 54 | 
            +
            	journal      = {GitHub repository},
         | 
| 55 | 
            +
            	publisher    = {GitHub},
         | 
| 56 | 
            +
            	howpublished = {\url{https://github.com/huggingface/trl}}
         | 
| 57 | 
            +
            }
         | 
| 58 | 
            +
            ```
         | 
    	
        added_tokens.json
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "<image_soft_token>": 262144
         | 
| 3 | 
            +
            }
         | 
    	
        chat_template.jinja
    ADDED
    
    | @@ -0,0 +1,47 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {{ bos_token }}
         | 
| 2 | 
            +
            {%- if messages[0]['role'] == 'system' -%}
         | 
| 3 | 
            +
                {%- if messages[0]['content'] is string -%}
         | 
| 4 | 
            +
                    {%- set first_user_prefix = messages[0]['content'] + '
         | 
| 5 | 
            +
             | 
| 6 | 
            +
            ' -%}
         | 
| 7 | 
            +
                {%- else -%}
         | 
| 8 | 
            +
                    {%- set first_user_prefix = messages[0]['content'][0]['text'] + '
         | 
| 9 | 
            +
             | 
| 10 | 
            +
            ' -%}
         | 
| 11 | 
            +
                {%- endif -%}
         | 
| 12 | 
            +
                {%- set loop_messages = messages[1:] -%}
         | 
| 13 | 
            +
            {%- else -%}
         | 
| 14 | 
            +
                {%- set first_user_prefix = "" -%}
         | 
| 15 | 
            +
                {%- set loop_messages = messages -%}
         | 
| 16 | 
            +
            {%- endif -%}
         | 
| 17 | 
            +
            {%- for message in loop_messages -%}
         | 
| 18 | 
            +
                {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
         | 
| 19 | 
            +
                    {{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
         | 
| 20 | 
            +
                {%- endif -%}
         | 
| 21 | 
            +
                {%- if (message['role'] == 'assistant') -%}
         | 
| 22 | 
            +
                    {%- set role = "model" -%}
         | 
| 23 | 
            +
                {%- else -%}
         | 
| 24 | 
            +
                    {%- set role = message['role'] -%}
         | 
| 25 | 
            +
                {%- endif -%}
         | 
| 26 | 
            +
                {{ '<start_of_turn>' + role + '
         | 
| 27 | 
            +
            ' + (first_user_prefix if loop.first else "") }}
         | 
| 28 | 
            +
                {%- if message['content'] is string -%}
         | 
| 29 | 
            +
                    {{ message['content'] | trim }}
         | 
| 30 | 
            +
                {%- elif message['content'] is iterable -%}
         | 
| 31 | 
            +
                    {%- for item in message['content'] -%}
         | 
| 32 | 
            +
                        {%- if item['type'] == 'image' -%}
         | 
| 33 | 
            +
                            {{ '<start_of_image>' }}
         | 
| 34 | 
            +
                        {%- elif item['type'] == 'text' -%}
         | 
| 35 | 
            +
                            {{ item['text'] | trim }}
         | 
| 36 | 
            +
                        {%- endif -%}
         | 
| 37 | 
            +
                    {%- endfor -%}
         | 
| 38 | 
            +
                {%- else -%}
         | 
| 39 | 
            +
                    {{ raise_exception("Invalid content type") }}
         | 
| 40 | 
            +
                {%- endif -%}
         | 
| 41 | 
            +
                {{ '<end_of_turn>
         | 
| 42 | 
            +
            ' }}
         | 
| 43 | 
            +
            {%- endfor -%}
         | 
| 44 | 
            +
            {%- if add_generation_prompt -%}
         | 
| 45 | 
            +
                {{'<start_of_turn>model
         | 
| 46 | 
            +
            '}}
         | 
| 47 | 
            +
            {%- endif -%}
         | 
    	
        config.json
    ADDED
    
    | @@ -0,0 +1,54 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "_sliding_window_pattern": 6,
         | 
| 3 | 
            +
              "architectures": [
         | 
| 4 | 
            +
                "Gemma3ForCausalLM"
         | 
| 5 | 
            +
              ],
         | 
| 6 | 
            +
              "attention_bias": false,
         | 
| 7 | 
            +
              "attention_dropout": 0.0,
         | 
| 8 | 
            +
              "attn_logit_softcapping": null,
         | 
| 9 | 
            +
              "bos_token_id": 2,
         | 
| 10 | 
            +
              "eos_token_id": 1,
         | 
| 11 | 
            +
              "final_logit_softcapping": null,
         | 
| 12 | 
            +
              "head_dim": 256,
         | 
| 13 | 
            +
              "hidden_activation": "gelu_pytorch_tanh",
         | 
| 14 | 
            +
              "hidden_size": 640,
         | 
| 15 | 
            +
              "initializer_range": 0.02,
         | 
| 16 | 
            +
              "intermediate_size": 2048,
         | 
| 17 | 
            +
              "layer_types": [
         | 
| 18 | 
            +
                "sliding_attention",
         | 
| 19 | 
            +
                "sliding_attention",
         | 
| 20 | 
            +
                "sliding_attention",
         | 
| 21 | 
            +
                "sliding_attention",
         | 
| 22 | 
            +
                "sliding_attention",
         | 
| 23 | 
            +
                "full_attention",
         | 
| 24 | 
            +
                "sliding_attention",
         | 
| 25 | 
            +
                "sliding_attention",
         | 
| 26 | 
            +
                "sliding_attention",
         | 
| 27 | 
            +
                "sliding_attention",
         | 
| 28 | 
            +
                "sliding_attention",
         | 
| 29 | 
            +
                "full_attention",
         | 
| 30 | 
            +
                "sliding_attention",
         | 
| 31 | 
            +
                "sliding_attention",
         | 
| 32 | 
            +
                "sliding_attention",
         | 
| 33 | 
            +
                "sliding_attention",
         | 
| 34 | 
            +
                "sliding_attention",
         | 
| 35 | 
            +
                "full_attention"
         | 
| 36 | 
            +
              ],
         | 
| 37 | 
            +
              "max_position_embeddings": 32768,
         | 
| 38 | 
            +
              "model_type": "gemma3_text",
         | 
| 39 | 
            +
              "num_attention_heads": 4,
         | 
| 40 | 
            +
              "num_hidden_layers": 18,
         | 
| 41 | 
            +
              "num_key_value_heads": 1,
         | 
| 42 | 
            +
              "pad_token_id": 0,
         | 
| 43 | 
            +
              "query_pre_attn_scalar": 256,
         | 
| 44 | 
            +
              "rms_norm_eps": 1e-06,
         | 
| 45 | 
            +
              "rope_local_base_freq": 10000.0,
         | 
| 46 | 
            +
              "rope_scaling": null,
         | 
| 47 | 
            +
              "rope_theta": 1000000.0,
         | 
| 48 | 
            +
              "sliding_window": 512,
         | 
| 49 | 
            +
              "torch_dtype": "bfloat16",
         | 
| 50 | 
            +
              "transformers_version": "4.55.0",
         | 
| 51 | 
            +
              "use_bidirectional_attention": false,
         | 
| 52 | 
            +
              "use_cache": true,
         | 
| 53 | 
            +
              "vocab_size": 262144
         | 
| 54 | 
            +
            }
         | 
    	
        model.safetensors
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:7833586c1e7e2a41ba630e6ae930f0e240e9a8b4f9c22c90fc08f36531e23619
         | 
| 3 | 
            +
            size 536223056
         | 
    	
        runs/Aug14_23-55-20_b500db30f9fe/events.out.tfevents.1755215734.b500db30f9fe.400.0
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:0b7f9a62b18dfad66736baa0c67eb615490f0d0035ab5e1edc02dbc147fed0a1
         | 
| 3 | 
            +
            size 10217
         | 
    	
        special_tokens_map.json
    ADDED
    
    | @@ -0,0 +1,33 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "boi_token": "<start_of_image>",
         | 
| 3 | 
            +
              "bos_token": {
         | 
| 4 | 
            +
                "content": "<bos>",
         | 
| 5 | 
            +
                "lstrip": false,
         | 
| 6 | 
            +
                "normalized": false,
         | 
| 7 | 
            +
                "rstrip": false,
         | 
| 8 | 
            +
                "single_word": false
         | 
| 9 | 
            +
              },
         | 
| 10 | 
            +
              "eoi_token": "<end_of_image>",
         | 
| 11 | 
            +
              "eos_token": {
         | 
| 12 | 
            +
                "content": "<eos>",
         | 
| 13 | 
            +
                "lstrip": false,
         | 
| 14 | 
            +
                "normalized": false,
         | 
| 15 | 
            +
                "rstrip": false,
         | 
| 16 | 
            +
                "single_word": false
         | 
| 17 | 
            +
              },
         | 
| 18 | 
            +
              "image_token": "<image_soft_token>",
         | 
| 19 | 
            +
              "pad_token": {
         | 
| 20 | 
            +
                "content": "<pad>",
         | 
| 21 | 
            +
                "lstrip": false,
         | 
| 22 | 
            +
                "normalized": false,
         | 
| 23 | 
            +
                "rstrip": false,
         | 
| 24 | 
            +
                "single_word": false
         | 
| 25 | 
            +
              },
         | 
| 26 | 
            +
              "unk_token": {
         | 
| 27 | 
            +
                "content": "<unk>",
         | 
| 28 | 
            +
                "lstrip": false,
         | 
| 29 | 
            +
                "normalized": false,
         | 
| 30 | 
            +
                "rstrip": false,
         | 
| 31 | 
            +
                "single_word": false
         | 
| 32 | 
            +
              }
         | 
| 33 | 
            +
            }
         | 
    	
        tokenizer.json
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
         | 
| 3 | 
            +
            size 33384568
         | 
    	
        tokenizer.model
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
         | 
| 3 | 
            +
            size 4689074
         | 
    	
        tokenizer_config.json
    ADDED
    
    | The diff for this file is too large to render. 
		See raw diff | 
|  | 
    	
        training_args.bin
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:cd0a092069a432f4a32ac50a860786d7bd20db39b7c9af1707e7e289a693b346
         | 
| 3 | 
            +
            size 5816
         | 
