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_21-55-01_pc/events.out.tfevents.1755233701.pc.47413.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: checkpoint_limero
         
     | 
| 5 | 
         
            +
            tags:
         
     | 
| 6 | 
         
            +
            - generated_from_trainer
         
     | 
| 7 | 
         
            +
            - sft
         
     | 
| 8 | 
         
            +
            - trl
         
     | 
| 9 | 
         
            +
            licence: license
         
     | 
| 10 | 
         
            +
            ---
         
     | 
| 11 | 
         
            +
             
     | 
| 12 | 
         
            +
            # Model Card for checkpoint_limero
         
     | 
| 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="crorella/checkpoint_limero", 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.2
         
     | 
| 39 | 
         
            +
            - Pytorch: 2.8.0
         
     | 
| 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.2",
         
     | 
| 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:77338db99447e13a933391904532512c60516afb3b01ee6f62fcea456e4cd612
         
     | 
| 3 | 
         
            +
            size 536223056
         
     | 
    	
        runs/Aug14_21-55-01_pc/events.out.tfevents.1755233701.pc.47413.0
    ADDED
    
    | 
         @@ -0,0 +1,3 @@ 
     | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 1 | 
         
            +
            version https://git-lfs.github.com/spec/v1
         
     | 
| 2 | 
         
            +
            oid sha256:021c39379c67806e50fdec38bd605501a98e552915e19a102a3653f63720aabd
         
     | 
| 3 | 
         
            +
            size 8251
         
     | 
    	
        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:d24fb8768f887117ac221a5413b41c3d25abfe9562a5684fa13792b564f059f8
         
     | 
| 3 | 
         
            +
            size 6225
         
     |