989f9a805638ba3fa756a0302dcba3bb74709a59dd1946b3bb05bf7ec07dac34

config.json

@@ -1,10 +1,11 @@
 {
-  "_name_or_path": "baichuan-inc/Baichuan-7B",
+  "_attn_implementation_autoset": true,
+  "_name_or_path": "/tmp/models/tmpphwma9oi/tmpyo9n14wg",
   "architectures": [
     "BaiChuanForCausalLM"
   ],
   "auto_map": {
-    "AutoConfig": "baichuan-inc/Baichuan-7B--configuration_baichuan.BaiChuanConfig",
+    "AutoConfig": "configuration_baichuan.BaiChuanConfig",
     "AutoModelForCausalLM": "baichuan-inc/Baichuan-7B--modeling_baichuan.BaiChuanForCausalLM"
   },
   "bos_token_id": 1,
@@ -18,6 +19,26 @@
   "num_attention_heads": 32,
   "num_hidden_layers": 32,
   "pad_token_id": 0,
+  "quantization_config": {
+    "quant_config": {
+      "offload_meta": false,
+      "scale_quant_params": null,
+      "weight_quant_params": {
+        "axis": 1,
+        "channel_wise": true,
+        "group_size": 64,
+        "nbits": 4,
+        "optimize": true,
+        "round_zero": true,
+        "view_as_float": false
+      },
+      "zero_quant_params": null
+    },
+    "quant_method": "hqq",
+    "skip_modules": [
+      "lm_head"
+    ]
+  },
   "rms_norm_eps": 1e-06,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",

configuration_baichuan.py

@@ -0,0 +1,66 @@
+# coding=utf-8
+# Copyright 2022 EleutherAI and the HuggingFace Inc. team. All rights reserved.
+#
+# This code is based on EleutherAI's GPT-NeoX library and the GPT-NeoX
+# and OPT implementations in this library. It has been modified from its
+# original forms to accommodate minor architectural differences compared
+# to GPT-NeoX and OPT used by the Meta AI team that trained the model.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from transformers.configuration_utils import PretrainedConfig
+from transformers.utils import logging
+
+
+logger = logging.get_logger(__name__)
+
+
+class BaiChuanConfig(PretrainedConfig):
+    model_type = "baichuan"
+    keys_to_ignore_at_inference = ["past_key_values"]
+
+    def __init__(
+        self,
+        vocab_size=64000,
+        hidden_size=4096,
+        intermediate_size=11008,
+        num_hidden_layers=32,
+        num_attention_heads=32,
+        hidden_act="silu",
+        max_position_embeddings=4096,
+        initializer_range=0.02,
+        rms_norm_eps=1e-6,
+        use_cache=True,
+        pad_token_id=0,
+        bos_token_id=1,
+        eos_token_id=2,
+        tie_word_embeddings=False,
+        **kwargs,
+    ):
+        self.vocab_size = vocab_size
+        self.max_position_embeddings = max_position_embeddings
+        self.hidden_size = hidden_size
+        self.intermediate_size = intermediate_size
+        self.num_hidden_layers = num_hidden_layers
+        self.num_attention_heads = num_attention_heads
+        self.hidden_act = hidden_act
+        self.initializer_range = initializer_range
+        self.rms_norm_eps = rms_norm_eps
+        self.use_cache = use_cache
+        super().__init__(
+            pad_token_id=pad_token_id,
+            bos_token_id=bos_token_id,
+            eos_token_id=eos_token_id,
+            tie_word_embeddings=tie_word_embeddings,
+            **kwargs,
+        )

smash_config.json

@@ -11,7 +11,7 @@
     "quant_hqq_weight_bits": 4,
     "max_batch_size": 1,
     "device": "cuda",
-    "cache_dir": "/tmp/models/tmp2p__e1si",
+    "cache_dir": "/tmp/models/tmpphwma9oi",
     "task": "",
     "save_load_fn": "hqq",
     "save_load_fn_args": {},