c71ba732a17d87fc62932cc1daba87e290385f702755d5646671271aa51b27cd

config.json

@@ -1,6 +1,6 @@
 {
   "_attn_implementation_autoset": true,
-  "_name_or_path": "berkeley-nest/Starling-LM-7B-alpha",
+  "_name_or_path": "/tmp/models/tmp6d9tk1n3/tmpw780lpp2",
   "architectures": [
     "MistralForCausalLM"
   ],
@@ -17,6 +17,26 @@
   "num_attention_heads": 32,
   "num_hidden_layers": 32,
   "num_key_value_heads": 8,
+  "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-05,
   "rope_theta": 10000.0,
   "sliding_window": 4096,

smash_config.json

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