End of training

README.md

@@ -0,0 +1,68 @@
+---
+tags:
+- generated_from_trainer
+model-index:
+- name: myBit-Llama2-jp-127M-test-16
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# myBit-Llama2-jp-127M-test-16
+
+This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 4.5029
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 0.00024
+- train_batch_size: 96
+- eval_batch_size: 96
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: polynomial
+- lr_scheduler_warmup_steps: 500
+- num_epochs: 1
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 9.0668        | 0.07  | 200  | 7.8313          |
+| 6.9502        | 0.15  | 400  | 6.3292          |
+| 5.8696        | 0.22  | 600  | 5.4250          |
+| 5.285         | 0.3   | 800  | 5.1157          |
+| 5.0167        | 0.37  | 1000 | 4.9604          |
+| 4.9441        | 0.45  | 1200 | 4.9913          |
+| 4.8204        | 0.52  | 1400 | 4.7712          |
+| 4.7314        | 0.6   | 1600 | 4.7058          |
+| 4.6562        | 0.67  | 1800 | 4.6503          |
+| 4.5941        | 0.74  | 2000 | 4.5901          |
+| 4.5479        | 0.82  | 2200 | 4.5508          |
+| 4.5155        | 0.89  | 2400 | 4.5194          |
+| 4.4939        | 0.97  | 2600 | 4.5029          |
+
+
+### Framework versions
+
+- Transformers 4.39.1
+- Pytorch 2.2.1+cu121
+- Datasets 2.18.0
+- Tokenizers 0.15.2

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "transformers_version": "4.39.1"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a418267732f701318275577ca358ff70e114727517671a25f4fd0ff702ff6b08
+oid sha256:abd0c9aba878cb57e408548148acfef4a39dfc2605abbabe634b69eee716459c
 size 511344824

modeling_bit_llama.py

@@ -0,0 +1,79 @@
+from typing import Optional
+from transformers.models.llama.modeling_llama import (
+    LlamaConfig, 
+    LlamaModel,
+    LlamaForCausalLM,
+    LlamaAttention,
+    LlamaFlashAttention2,
+    LlamaSdpaAttention,
+    LlamaMLP,
+    LlamaDecoderLayer,
+)
+from mybitnet.bitnet import BitLinear
+from torch import nn
+
+class BitLlamaConfig(LlamaConfig):
+    model_type = "bit_llama"
+
+    def __init__(self, bits=8, **kwargs):
+        super().__init__(**kwargs)
+        self.bits = bits
+
+class BitLlamaMLP(LlamaMLP):
+    def __init__(self, config):
+        super().__init__(config)
+        self.gate_proj = BitLinear(self.hidden_size, self.intermediate_size, bias=False, bits=config.bits, flg_before_linear=True)
+        self.up_proj = BitLinear(self.hidden_size, self.intermediate_size, bias=False, bits=config.bits, flg_before_linear=True)
+        self.down_proj = BitLinear(self.intermediate_size, self.hidden_size, bias=False, bits=config.bits, flg_before_linear=False)
+        
+class BitLlamaAttention(LlamaAttention):
+    def __init__(self, config: BitLlamaConfig, layer_idx: Optional[int] = None):
+        super().__init__(config)
+        self.q_proj = BitLinear(self.hidden_size, self.num_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.k_proj = BitLinear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.v_proj = BitLinear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.o_proj = BitLinear(self.hidden_size, self.hidden_size, bias=False, bits=config.bits, flg_before_linear=True)
+
+class BitLlamaFlashAttention2(LlamaFlashAttention2):
+    def __init__(self, config: BitLlamaConfig, layer_idx: Optional[int] = None):
+        super().__init__(config, layer_idx)
+        self.q_proj = BitLinear(self.hidden_size, self.num_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.k_proj = BitLinear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.v_proj = BitLinear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.o_proj = BitLinear(self.hidden_size, self.hidden_size, bias=False, bits=config.bits, flg_before_linear=True)
+
+class BitLlamaSdpaAttention(LlamaSdpaAttention):
+    def __init__(self, config: BitLlamaConfig, layer_idx: Optional[int] = None):
+        super().__init__(config, layer_idx)
+        self.q_proj = BitLinear(self.hidden_size, self.num_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.k_proj = BitLinear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.v_proj = BitLinear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=False, bits=config.bits, flg_before_linear=True)
+        self.o_proj = BitLinear(self.hidden_size, self.hidden_size, bias=False, bits=config.bits, flg_before_linear=True)
+
+BITLLAMA_ATTENTION_CLASSES = {
+    "eager": BitLlamaAttention,
+    "flash_attention_2": BitLlamaFlashAttention2,
+    "sdpa": BitLlamaSdpaAttention,
+}
+
+class BitLlamaDecoderLayer(LlamaDecoderLayer):
+    def __init__(self, config: BitLlamaConfig, layer_idx: int):
+        super().__init__(config, layer_idx)
+        self.self_attn = BITLLAMA_ATTENTION_CLASSES[config._attn_implementation](config=config, layer_idx=layer_idx)
+        self.mlp = BitLlamaMLP(config)
+
+class BitLlamaModel(LlamaModel):
+    def __init__(self, config: BitLlamaConfig):
+        super().__init__(config)
+        self.layers = nn.ModuleList(
+            [BitLlamaDecoderLayer(config, layer_idx) for layer_idx in range(config.num_hidden_layers)]
+        )
+
+class BitLlamaForCausalLM(LlamaForCausalLM):
+    config_class = BitLlamaConfig
+
+    def __init__(self, config: BitLlamaConfig):
+        super().__init__(config)
+        self.model = BitLlamaModel(config)
+        self.lm_head = BitLinear(config.hidden_size, config.vocab_size, bias=False, bits=config.bits, flg_before_linear=True)
+