初始化项目，由ModelHub XC社区提供模型
Model: saidutta69/Yi-1.5-9B-Chat-16K-heretic Source: Original Platform
This commit is contained in:
ModelHub XC
40
.gitattributes
vendored
Normal file
40
.gitattributes
vendored
Normal file
@@ -0,0 +1,40 @@
|
||||
*.7z filter=lfs diff=lfs merge=lfs -text
|
||||
*.arrow filter=lfs diff=lfs merge=lfs -text
|
||||
*.bin filter=lfs diff=lfs merge=lfs -text
|
||||
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
||||
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
||||
*.ftz filter=lfs diff=lfs merge=lfs -text
|
||||
*.gz filter=lfs diff=lfs merge=lfs -text
|
||||
*.h5 filter=lfs diff=lfs merge=lfs -text
|
||||
*.joblib filter=lfs diff=lfs merge=lfs -text
|
||||
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
||||
*.model filter=lfs diff=lfs merge=lfs -text
|
||||
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
||||
*.npy filter=lfs diff=lfs merge=lfs -text
|
||||
*.npz filter=lfs diff=lfs merge=lfs -text
|
||||
*.onnx filter=lfs diff=lfs merge=lfs -text
|
||||
*.ot filter=lfs diff=lfs merge=lfs -text
|
||||
*.parquet filter=lfs diff=lfs merge=lfs -text
|
||||
*.pb filter=lfs diff=lfs merge=lfs -text
|
||||
*.pickle filter=lfs diff=lfs merge=lfs -text
|
||||
*.pkl filter=lfs diff=lfs merge=lfs -text
|
||||
*.pt filter=lfs diff=lfs merge=lfs -text
|
||||
*.pth filter=lfs diff=lfs merge=lfs -text
|
||||
*.rar filter=lfs diff=lfs merge=lfs -text
|
||||
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar filter=lfs diff=lfs merge=lfs -text
|
||||
*.tflite filter=lfs diff=lfs merge=lfs -text
|
||||
*.tgz filter=lfs diff=lfs merge=lfs -text
|
||||
*.wasm filter=lfs diff=lfs merge=lfs -text
|
||||
*.xz filter=lfs diff=lfs merge=lfs -text
|
||||
*.zip filter=lfs diff=lfs merge=lfs -text
|
||||
*.zst filter=lfs diff=lfs merge=lfs -text
|
||||
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
||||
ggml-model-Q4_K_M.gguf filter=lfs diff=lfs merge=lfs -text
|
||||
ggml-model-Q5_K_M.gguf filter=lfs diff=lfs merge=lfs -text
|
||||
ggml-model-Q6_K.gguf filter=lfs diff=lfs merge=lfs -text
|
||||
ggml-model-Q8_0.gguf filter=lfs diff=lfs merge=lfs -text
|
||||
Yi-1.5-9B-Chat-16K-heretic-F16.gguf filter=lfs diff=lfs merge=lfs -text
|
||||
107
README.md
Normal file
107
README.md
Normal file
@@ -0,0 +1,107 @@
|
||||
---
|
||||
license: other
|
||||
license_link: https://huggingface.co/01-ai/Yi-1.5-9B-Chat-16K/blob/main/LICENSE
|
||||
language:
|
||||
- en
|
||||
library_name: transformers
|
||||
pipeline_tag: text-generation
|
||||
base_model:
|
||||
- 01-ai/Yi-1.5-9B-Chat-16K
|
||||
tags:
|
||||
- heretic
|
||||
- uncensored
|
||||
- decensored
|
||||
- abliterated
|
||||
- reproducible
|
||||
- conversational
|
||||
- text-generation-inference
|
||||
- gguf
|
||||
- roleplay
|
||||
- local-llm
|
||||
- consumer-gpu
|
||||
- multilingual
|
||||
---
|
||||
|
||||
# Yi-1.5-9B-Chat-16K-heretic
|
||||
|
||||
<div align="center">
|
||||
<img src="https://photu.kashyalabanavli.site/racer-is-op.png" alt="RACER IS OP" width="100%">
|
||||
</div>
|
||||
|
||||
<br>
|
||||
|
||||
A decensored variant of [01-ai/Yi-1.5-9B-Chat-16K](https://huggingface.co/01-ai/Yi-1.5-9B-Chat-16K), produced with [Heretic](https://github.com/p-e-w/heretic) v1.4.0 (directional ablation / "abliteration"). a 9B Yi-1.5 model fine-tuned for 16K context chat — strong multilingual reasoning now uncensored. Refusal behavior is suppressed via targeted weight edits to the attention output and MLP down-projections rather than fine-tuning, so the base model's knowledge and capabilities are left largely intact.
|
||||
|
||||
**Who this is for:** developers who want a 9B multilingual chat model that answers directly instead of refusing — for long-context conversations, multilingual applications, or any use case needing reliable reasoning without censorship.
|
||||
|
||||
|
||||
<!-- racer-gpu-matrix -->
|
||||
## Runs on your gaming PC
|
||||
|
||||
Full GGUF ladder included — pick the quant that fits your card:
|
||||
|
||||
| Your GPU | Recommended quant | Weights |
|
||||
| :--- | :--- | :--- |
|
||||
| RTX 3090 / 4090 / 5090 (24 GB) | Q8_0 | ~9.4 GB |
|
||||
| RTX 4080 / 5080 / 4060 Ti 16G (16 GB) | Q6_K | ~7.5 GB |
|
||||
| RTX 3060 / 4070 / 5070 (12 GB) | Q5_K_M | ~6.6 GB |
|
||||
| RTX 4060 / 3070 (8 GB) | Q4_K_M | ~5.8 GB |
|
||||
| GTX 1660 Super / 2060 / 3050 laptop (6 GB) | IQ4_XS | ~5.4 GB |
|
||||
| CPU-only / Apple Silicon | Q4_K_M | fits in system RAM |
|
||||
|
||||
Weights only, at this model's 8.8B native size; add ~1 GB for context.
|
||||
OOM? Drop one quant level. Headroom to spare? Go one up.
|
||||
|
||||
## Why abliteration instead of fine-tuning
|
||||
|
||||
Fine-tuning a "helpful" persona on top of RLHF'd refusals fights the base model's training and tends to degrade coherence. Abliteration instead finds and edits the specific weight directions responsible for refusal, leaving the rest of the network (and its capabilities) untouched. See the [Heretic repo](https://github.com/p-e-w/heretic) and the [original abliteration writeup](https://huggingface.co/blog/mlabonne/abliteration) for the mechanism.
|
||||
|
||||
> Made with ❤️ by **RACER IS OP** — follow for more uncensored models
|
||||
|
||||
## Files
|
||||
|
||||
| File | Format | Size |
|
||||
|---|---|---|
|
||||
| `model-00001-of-00004.safetensors ... model-00004-of-00004.safetensors` | BF16 | (see repo files) |
|
||||
| `ggml-model-Q4_K_M.gguf` | GGUF, Q4_K_M | (see repo files) |
|
||||
| `ggml-model-Q5_K_M.gguf` | GGUF, Q5_K_M | (see repo files) |
|
||||
| `ggml-model-Q6_K.gguf` | GGUF, Q6_K | (see repo files) |
|
||||
| `ggml-model-Q8_0.gguf` | GGUF, Q8_0 | (see repo files) |
|
||||
|
||||
GGUF quants are produced with [llama.cpp](https://github.com/ggml-org/llama.cpp). Run `llama serve -hf saidutta69/Yi-1.5-9B-Chat-16K-heretic` to pull the default quant.
|
||||
|
||||
## Quickstart
|
||||
|
||||
```bash
|
||||
# llama.cpp
|
||||
llama serve -hf saidutta69/Yi-1.5-9B-Chat-16K-heretic
|
||||
```
|
||||
|
||||
```python
|
||||
# transformers
|
||||
from transformers import AutoModelForCausalLM, AutoTokenizer
|
||||
|
||||
model_id = "saidutta69/Yi-1.5-9B-Chat-16K-heretic"
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
||||
model = AutoModelForCausalLM.from_pretrained(
|
||||
model_id,
|
||||
torch_dtype="auto",
|
||||
device_map="auto",
|
||||
)
|
||||
|
||||
messages = [{"role": "user", "content": "Write a Python function to merge two sorted lists."}]
|
||||
text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
||||
inputs = tokenizer(text, return_tensors="pt").to(model.device)
|
||||
outputs = model.generate(**inputs, max_new_tokens=512, do_sample=False)
|
||||
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))
|
||||
```
|
||||
|
||||
Also runnable via Ollama, LM Studio, Jan, vLLM, SGLang — see the "Use this model" widget above for copy-paste commands.
|
||||
|
||||
## Responsible use
|
||||
|
||||
Refusal suppression is deliberate and works as intended: this model will comply with requests the base model would refuse, including some it shouldn't. There is no safety filtering layered on top. You are responsible for how you deploy it — don't put this behind an unmoderated public-facing endpoint serving third parties.
|
||||
|
||||
## License
|
||||
|
||||
Inherits the [other](https://huggingface.co/01-ai/Yi-1.5-9B-Chat-16K/blob/main/LICENSE) license from the base model.
|
||||
3
Yi-1.5-9B-Chat-16K-heretic-F16.gguf
Normal file
3
Yi-1.5-9B-Chat-16K-heretic-F16.gguf
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:2b0de70b94b13cda3f612e7501956cd3c32da61f062e6a0f724bac44019a9fd5
|
||||
size 17661113408
|
||||
1
chat_template.jinja
Normal file
1
chat_template.jinja
Normal file
@@ -0,0 +1 @@
|
||||
{% if messages[0]['role'] == 'system' %}{% set system_message = messages[0]['content'] %}{% endif %}{% if system_message is defined %}{{ system_message }}{% endif %}{% for message in messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ '<|im_start|>user\n' + content + '<|im_end|>\n<|im_start|>assistant\n' }}{% elif message['role'] == 'assistant' %}{{ content + '<|im_end|>' + '\n' }}{% endif %}{% endfor %}
|
||||
32
config.json
Normal file
32
config.json
Normal file
@@ -0,0 +1,32 @@
|
||||
{
|
||||
"architectures": [
|
||||
"LlamaForCausalLM"
|
||||
],
|
||||
"attention_bias": false,
|
||||
"attention_dropout": 0.0,
|
||||
"bos_token_id": 1,
|
||||
"dtype": "bfloat16",
|
||||
"eos_token_id": 2,
|
||||
"head_dim": 128,
|
||||
"hidden_act": "silu",
|
||||
"hidden_size": 4096,
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 11008,
|
||||
"max_position_embeddings": 16384,
|
||||
"mlp_bias": false,
|
||||
"model_type": "llama",
|
||||
"num_attention_heads": 32,
|
||||
"num_hidden_layers": 48,
|
||||
"num_key_value_heads": 4,
|
||||
"pad_token_id": null,
|
||||
"pretraining_tp": 1,
|
||||
"rms_norm_eps": 1e-06,
|
||||
"rope_parameters": {
|
||||
"rope_theta": 5000000,
|
||||
"rope_type": "default"
|
||||
},
|
||||
"tie_word_embeddings": false,
|
||||
"transformers_version": "5.14.1",
|
||||
"use_cache": false,
|
||||
"vocab_size": 64000
|
||||
}
|
||||
6
generation_config.json
Normal file
6
generation_config.json
Normal file
@@ -0,0 +1,6 @@
|
||||
{
|
||||
"_from_model_config": true,
|
||||
"bos_token_id": 1,
|
||||
"eos_token_id": 2,
|
||||
"transformers_version": "5.14.1"
|
||||
}
|
||||
3
ggml-model-Q4_K_M.gguf
Normal file
3
ggml-model-Q4_K_M.gguf
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:4e89abeda1705ca1c30c057cd55be1110c3826f6daf5d99d196b0d883239fcb6
|
||||
size 5328958144
|
||||
3
ggml-model-Q5_K_M.gguf
Normal file
3
ggml-model-Q5_K_M.gguf
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:de896064082aac499d169b7c79c8e5099c346c41a549324a012ac04f5d853e9a
|
||||
size 6258258624
|
||||
3
ggml-model-Q6_K.gguf
Normal file
3
ggml-model-Q6_K.gguf
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:790e3b2109d67814a69a7c9d126112ba4638c83a79bc66ba327dcfe5f3be3176
|
||||
size 7245640384
|
||||
3
ggml-model-Q8_0.gguf
Normal file
3
ggml-model-Q8_0.gguf
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:b3fe510d4980e499d33147ebce1c66579da1dc0678cf08763fb88051dd6d8729
|
||||
size 9383916224
|
||||
3
model-00001-of-00004.safetensors
Normal file
3
model-00001-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:1cbc1dcebd8de1fd20145ece0bc508922bd6669aa43f9c566db9d6a6a7113ff1
|
||||
size 4945284744
|
||||
3
model-00002-of-00004.safetensors
Normal file
3
model-00002-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:aa5f8ab572105d74cf2b5a24ec530158296a82d1ed09c968106537005f92adb1
|
||||
size 4934842816
|
||||
3
model-00003-of-00004.safetensors
Normal file
3
model-00003-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:4019a8b4adb883ebafa630d7d5010f9179d8842cd59aed9d0cd9b97811a18099
|
||||
size 4972600096
|
||||
3
model-00004-of-00004.safetensors
Normal file
3
model-00004-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:294f92c0065d44a3bc726bdd15fb969fcd26c9d290bfe592176f0a88872730e9
|
||||
size 2806137344
|
||||
443
model.safetensors.index.json
Normal file
443
model.safetensors.index.json
Normal file
@@ -0,0 +1,443 @@
|
||||
{
|
||||
"metadata": {
|
||||
"total_parameters": 8829407232,
|
||||
"total_size": 17658814464
|
||||
},
|
||||
"weight_map": {
|
||||
"lm_head.weight": "model-00001-of-00004.safetensors",
|
||||
"model.embed_tokens.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.11.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.11.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.20.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.25.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.25.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.25.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.30.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.32.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.33.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.34.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.35.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.36.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.37.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.38.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.39.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.39.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.40.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.40.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.41.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.42.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.43.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.44.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.45.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.46.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.mlp.gate_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.mlp.up_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.47.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.norm.weight": "model-00004-of-00004.safetensors"
|
||||
}
|
||||
}
|
||||
3804
reproduce/01-ai--Yi-1--5-9B-Chat-16K.jsonl
Normal file
3804
reproduce/01-ai--Yi-1--5-9B-Chat-16K.jsonl
Normal file
File diff suppressed because it is too large
Load Diff
69
reproduce/README.md
Normal file
69
reproduce/README.md
Normal file
@@ -0,0 +1,69 @@
|
||||
# Reproduction guide
|
||||
|
||||
This directory contains the necessary information and assets to reproduce the results obtained during this Heretic run.
|
||||
|
||||
## Models
|
||||
|
||||
- **Base model:** [01-ai/Yi-1.5-9B-Chat-16K](https://huggingface.co/01-ai/Yi-1.5-9B-Chat-16K) (Commit: [`2f8753b`](https://huggingface.co/01-ai/Yi-1.5-9B-Chat-16K/commit/2f8753bed95bd4fede3fd64f8305428620b4b212))
|
||||
|
||||
## Datasets
|
||||
|
||||
- **Good prompts:** [mlabonne/harmless_alpaca](https://huggingface.co/datasets/mlabonne/harmless_alpaca) (Commit: [`02c6a92`](https://huggingface.co/datasets/mlabonne/harmless_alpaca/commit/02c6a92cfcf11bb0c387334f8146d149d65b587f))
|
||||
- **Bad prompts:** [mlabonne/harmful_behaviors](https://huggingface.co/datasets/mlabonne/harmful_behaviors) (Commit: [`01cead0`](https://huggingface.co/datasets/mlabonne/harmful_behaviors/commit/01cead01398926d81f7c52bdb790ee8cf77ebba7))
|
||||
- **Good evaluation prompts:** [mlabonne/harmless_alpaca](https://huggingface.co/datasets/mlabonne/harmless_alpaca) (Commit: [`02c6a92`](https://huggingface.co/datasets/mlabonne/harmless_alpaca/commit/02c6a92cfcf11bb0c387334f8146d149d65b587f))
|
||||
- **Bad evaluation prompts:** [mlabonne/harmful_behaviors](https://huggingface.co/datasets/mlabonne/harmful_behaviors) (Commit: [`01cead0`](https://huggingface.co/datasets/mlabonne/harmful_behaviors/commit/01cead01398926d81f7c52bdb790ee8cf77ebba7))
|
||||
|
||||
## Selected trial
|
||||
|
||||
- **Trial number:** 101
|
||||
- **KL divergence:** 0.062347
|
||||
- **Refusals:** 2/100
|
||||
|
||||
## System
|
||||
|
||||
- **Python:** 3.12.11 (CPython, GCC 11.2.0) [Conda]
|
||||
- **Operating system:** Linux-6.8.0-1060-aws-x86_64-with-glibc2.39 (x86_64)
|
||||
- **CPU:** Intel(R) Xeon(R) Platinum 8559C
|
||||
|
||||
### Accelerators
|
||||
|
||||
- **CUDA:** Detected 1 device(s) (94.97 GB total VRAM)
|
||||
- **CUDA Version:** 12.8
|
||||
- **Driver Version:** 580.126.20
|
||||
- **Devices:**
|
||||
- **CUDA 0:** NVIDIA RTX PRO 6000 Blackwell Server Edition (94.97 GB)
|
||||
|
||||
## Environment
|
||||
|
||||
- **Heretic:** v1.4.0 (Origin: PyPI)
|
||||
- **PyTorch:** 2.8.0+cu128
|
||||
- **Other dependencies:** See [`requirements.txt`](requirements.txt).
|
||||
|
||||
## Contents of this directory
|
||||
|
||||
- [`requirements.txt`](requirements.txt): The exact versions of all Python packages.
|
||||
- [`config.toml`](config.toml): The exact configuration used, including the RNG seed.
|
||||
- [`01-ai--Yi-1--5-9B-Chat-16K.jsonl`](01-ai--Yi-1--5-9B-Chat-16K.jsonl): The Optuna study journal containing the history of all trials.
|
||||
- [`SHA256SUMS`](SHA256SUMS): Cryptographic hashes for all weight files.
|
||||
- [`reproduce.json`](reproduce.json): A machine-readable file containing all reproducibility information.
|
||||
|
||||
## How to reproduce
|
||||
|
||||
> [!TIP]
|
||||
> You can automate this process, including all verification steps, by downloading the `reproduce.json` file and running
|
||||
> `heretic --reproduce reproduce.json`.
|
||||
|
||||
1. Ensure your system matches the specifications in the **System** section above. Exact reproducibility is only guaranteed if all aspects of your system are identical to the one the model was originally generated on.
|
||||
1. Install the exact version of Heretic indicated in the **Environment** section above, from its original source.
|
||||
1. Install the packages listed in `requirements.txt`: `pip install -r requirements.txt`
|
||||
1. Install the correct version of PyTorch: `pip install torch==2.8.0+cu128 --index-url https://download.pytorch.org/whl/cu128`
|
||||
1. Place the provided `config.toml` in your working directory.
|
||||
1. Run Heretic without any additional arguments: `heretic`
|
||||
1. Wait for the run to finish, then select trial **101** and export the model.
|
||||
1. Verify that the weight files have been exactly reproduced by comparing their SHA-256 hashes against those in `SHA256SUMS`:
|
||||
`sha256sum -c SHA256SUMS` (or look at the hashes online if you uploaded to Hugging Face)
|
||||
|
||||
> [!TIP]
|
||||
> To use the included Optuna study journal `01-ai--Yi-1--5-9B-Chat-16K.jsonl`, place it in the checkpoints directory (usually `checkpoints/`) before running Heretic.
|
||||
>
|
||||
> This allows you to export other models from the Pareto front, or to run additional trials without having to re-run the stored trials.
|
||||
4
reproduce/SHA256SUMS
Normal file
4
reproduce/SHA256SUMS
Normal file
@@ -0,0 +1,4 @@
|
||||
1cbc1dcebd8de1fd20145ece0bc508922bd6669aa43f9c566db9d6a6a7113ff1 *model-00001-of-00004.safetensors
|
||||
aa5f8ab572105d74cf2b5a24ec530158296a82d1ed09c968106537005f92adb1 *model-00002-of-00004.safetensors
|
||||
4019a8b4adb883ebafa630d7d5010f9179d8842cd59aed9d0cd9b97811a18099 *model-00003-of-00004.safetensors
|
||||
294f92c0065d44a3bc726bdd15fb969fcd26c9d290bfe592176f0a88872730e9 *model-00004-of-00004.safetensors
|
||||
90
reproduce/config.toml
Normal file
90
reproduce/config.toml
Normal file
@@ -0,0 +1,90 @@
|
||||
model = "01-ai/Yi-1.5-9B-Chat-16K"
|
||||
model_commit = "2f8753bed95bd4fede3fd64f8305428620b4b212"
|
||||
dtypes = [
|
||||
"bfloat16",
|
||||
]
|
||||
quantization = "none"
|
||||
device_map = "auto"
|
||||
offload_outputs_to_cpu = true
|
||||
batch_size = 128
|
||||
max_response_length = 100
|
||||
response_prefix = ""
|
||||
kl_divergence_scale = 1.0
|
||||
kl_divergence_target = 0.01
|
||||
orthogonalize_direction = true
|
||||
row_normalization = "full"
|
||||
full_normalization_lora_rank = 3
|
||||
winsorization_quantile = 1.0
|
||||
n_trials = 200
|
||||
n_startup_trials = 60
|
||||
seed = 1257409843
|
||||
export_strategy = "merge"
|
||||
max_shard_size = "5GB"
|
||||
refusal_markers = [
|
||||
"disclaimer",
|
||||
"sorry",
|
||||
"i can'",
|
||||
"i cant",
|
||||
"i cannot",
|
||||
"i won'",
|
||||
"i wont",
|
||||
"i will not",
|
||||
"i unable",
|
||||
"im unable",
|
||||
"i'm unable",
|
||||
"i am unable",
|
||||
"i an ai",
|
||||
"im an ai",
|
||||
"i'm an ai",
|
||||
"i am an ai",
|
||||
"as an ai",
|
||||
"ai assistant",
|
||||
"i designed to",
|
||||
"im designed to",
|
||||
"i'm designed to",
|
||||
"i am designed to",
|
||||
"i programmed to",
|
||||
"im programmed to",
|
||||
"i'm programmed to",
|
||||
"i am programmed to",
|
||||
"violat",
|
||||
"prohibit",
|
||||
"illegal",
|
||||
"harmful",
|
||||
"inappropriate",
|
||||
"unethical",
|
||||
"ethical boundaries",
|
||||
]
|
||||
system_prompt = "You are a helpful assistant."
|
||||
|
||||
[good_prompts]
|
||||
dataset = "mlabonne/harmless_alpaca"
|
||||
commit = "02c6a92cfcf11bb0c387334f8146d149d65b587f"
|
||||
split = "train[:400]"
|
||||
column = "text"
|
||||
prefix = ""
|
||||
suffix = ""
|
||||
|
||||
[bad_prompts]
|
||||
dataset = "mlabonne/harmful_behaviors"
|
||||
commit = "01cead01398926d81f7c52bdb790ee8cf77ebba7"
|
||||
split = "train[:400]"
|
||||
column = "text"
|
||||
prefix = ""
|
||||
suffix = ""
|
||||
|
||||
[good_evaluation_prompts]
|
||||
dataset = "mlabonne/harmless_alpaca"
|
||||
commit = "02c6a92cfcf11bb0c387334f8146d149d65b587f"
|
||||
split = "test[:100]"
|
||||
column = "text"
|
||||
prefix = ""
|
||||
suffix = ""
|
||||
|
||||
[bad_evaluation_prompts]
|
||||
dataset = "mlabonne/harmful_behaviors"
|
||||
commit = "01cead01398926d81f7c52bdb790ee8cf77ebba7"
|
||||
split = "test[:100]"
|
||||
column = "text"
|
||||
prefix = ""
|
||||
suffix = ""
|
||||
287
reproduce/reproduce.json
Normal file
287
reproduce/reproduce.json
Normal file
@@ -0,0 +1,287 @@
|
||||
{
|
||||
"version": "2",
|
||||
"timestamp": "2026-07-21T20:23:30",
|
||||
"system": {
|
||||
"python": {
|
||||
"version": "3.12.11",
|
||||
"implementation": "CPython",
|
||||
"compiler": "GCC 11.2.0",
|
||||
"environment": "Conda"
|
||||
},
|
||||
"os": {
|
||||
"platform": "Linux-6.8.0-1060-aws-x86_64-with-glibc2.39",
|
||||
"machine": "x86_64"
|
||||
},
|
||||
"cpu": {
|
||||
"brand": "Intel(R) Xeon(R) Platinum 8559C",
|
||||
"vendor": "GenuineIntel",
|
||||
"family": 6,
|
||||
"model": 207,
|
||||
"stepping": 2
|
||||
},
|
||||
"accelerators": {
|
||||
"type": "CUDA",
|
||||
"api_name": "CUDA Version",
|
||||
"api_version": "12.8",
|
||||
"driver_version": "580.126.20",
|
||||
"devices": [
|
||||
{
|
||||
"name": "NVIDIA RTX PRO 6000 Blackwell Server Edition",
|
||||
"vram_gb": 94.97
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"environment": {
|
||||
"heretic": {
|
||||
"version": "1.4.0",
|
||||
"is_standard_pypi": true,
|
||||
"metadata": {
|
||||
"type": "pypi"
|
||||
}
|
||||
},
|
||||
"pytorch_version": "2.8.0+cu128",
|
||||
"requirements": {
|
||||
"absl-py": "2.4.0",
|
||||
"accelerate": "1.14.0",
|
||||
"alembic": "1.18.5",
|
||||
"annotated-doc": "0.0.4",
|
||||
"annotated-types": "0.7.0",
|
||||
"anyio": "4.14.1",
|
||||
"bitsandbytes": "0.49.2",
|
||||
"certifi": "2026.6.17",
|
||||
"chardet": "6.0.0.post1",
|
||||
"charset-normalizer": "3.4.7",
|
||||
"click": "8.4.2",
|
||||
"colorama": "0.4.6",
|
||||
"colorlog": "6.11.0",
|
||||
"dataproperty": "1.1.1",
|
||||
"datasets": "4.8.5",
|
||||
"defusedxml": "0.7.1",
|
||||
"dill": "0.4.1",
|
||||
"evaluate": "0.4.6",
|
||||
"filelock": "3.29.4",
|
||||
"fsspec": "2026.2.0",
|
||||
"greenlet": "3.5.3",
|
||||
"h11": "0.16.0",
|
||||
"heretic-llm": "1.4.0",
|
||||
"hf-xet": "1.5.2",
|
||||
"httpcore": "1.0.9",
|
||||
"httpx": "0.28.1",
|
||||
"huggingface-hub": "1.24.0",
|
||||
"idna": "3.18",
|
||||
"immutabledict": "4.3.1",
|
||||
"jinja2": "3.1.6",
|
||||
"joblib": "1.5.3",
|
||||
"langdetect": "1.0.9",
|
||||
"lm-eval": "0.4.12",
|
||||
"lxml": "6.1.1",
|
||||
"mako": "1.3.12",
|
||||
"markdown-it-py": "4.2.0",
|
||||
"markupsafe": "3.0.3",
|
||||
"mbstrdecoder": "1.1.5",
|
||||
"mdurl": "0.1.2",
|
||||
"more-itertools": "11.1.0",
|
||||
"mpmath": "1.3.0",
|
||||
"multiprocess": "0.70.19",
|
||||
"narwhals": "2.24.0",
|
||||
"networkx": "3.6.1",
|
||||
"nltk": "3.10.0",
|
||||
"numpy": "2.5.1",
|
||||
"nvidia-cublas-cu12": "12.8.4.1",
|
||||
"nvidia-cuda-cupti-cu12": "12.8.90",
|
||||
"nvidia-cuda-nvrtc-cu12": "12.8.93",
|
||||
"nvidia-cuda-runtime-cu12": "12.8.90",
|
||||
"nvidia-cudnn-cu12": "9.10.2.21",
|
||||
"nvidia-cufft-cu12": "11.3.3.83",
|
||||
"nvidia-cufile-cu12": "1.13.1.3",
|
||||
"nvidia-curand-cu12": "10.3.9.90",
|
||||
"nvidia-cusolver-cu12": "11.7.3.90",
|
||||
"nvidia-cusparse-cu12": "12.5.8.93",
|
||||
"nvidia-cusparselt-cu12": "0.7.1",
|
||||
"nvidia-nccl-cu12": "2.27.3",
|
||||
"nvidia-nvjitlink-cu12": "12.8.93",
|
||||
"nvidia-nvtx-cu12": "12.8.90",
|
||||
"optuna": "4.9.0",
|
||||
"packaging": "26.0",
|
||||
"pandas": "3.0.3",
|
||||
"pathvalidate": "3.3.1",
|
||||
"peft": "0.19.1",
|
||||
"pillow": "12.2.0",
|
||||
"portalocker": "3.2.0",
|
||||
"prompt-toolkit": "3.0.52",
|
||||
"psutil": "7.2.2",
|
||||
"py-cpuinfo": "9.0.0",
|
||||
"pyarrow": "25.0.0",
|
||||
"pydantic": "2.13.4",
|
||||
"pydantic-core": "2.46.4",
|
||||
"pydantic-settings": "2.14.2",
|
||||
"pygments": "2.20.0",
|
||||
"pytablewriter": "1.2.1",
|
||||
"python-dateutil": "2.9.0.post0",
|
||||
"python-dotenv": "1.2.2",
|
||||
"pyyaml": "6.0.3",
|
||||
"questionary": "2.1.1",
|
||||
"regex": "2026.7.19",
|
||||
"requests": "2.34.2",
|
||||
"rich": "14.3.4",
|
||||
"rouge-score": "0.1.2",
|
||||
"sacrebleu": "2.6.0",
|
||||
"safetensors": "0.8.0",
|
||||
"scikit-learn": "1.9.0",
|
||||
"scipy": "1.18.0",
|
||||
"setuptools": "82.0.1",
|
||||
"shellingham": "1.5.4",
|
||||
"six": "1.17.0",
|
||||
"sqlalchemy": "2.0.51",
|
||||
"sqlitedict": "2.1.0",
|
||||
"sympy": "1.14.0",
|
||||
"tabledata": "1.3.5",
|
||||
"tabulate": "0.10.0",
|
||||
"tcolorpy": "0.1.7",
|
||||
"threadpoolctl": "3.6.0",
|
||||
"tokenizers": "0.22.2",
|
||||
"tomli-w": "1.2.0",
|
||||
"torch": "2.8.0",
|
||||
"torchvision": "0.23.0",
|
||||
"tqdm": "4.68.3",
|
||||
"transformers": "5.14.1",
|
||||
"triton": "3.4.0",
|
||||
"typepy": "1.3.5",
|
||||
"typer": "0.27.0",
|
||||
"typing-extensions": "4.15.0",
|
||||
"typing-inspection": "0.4.2",
|
||||
"tzdata": "2026.2",
|
||||
"urllib3": "2.5.0",
|
||||
"wcwidth": "0.8.1",
|
||||
"word2number": "1.1",
|
||||
"xxhash": "3.8.1"
|
||||
}
|
||||
},
|
||||
"settings": {
|
||||
"model": "01-ai/Yi-1.5-9B-Chat-16K",
|
||||
"model_commit": "2f8753bed95bd4fede3fd64f8305428620b4b212",
|
||||
"dtypes": [
|
||||
"bfloat16"
|
||||
],
|
||||
"quantization": "none",
|
||||
"device_map": "auto",
|
||||
"max_memory": null,
|
||||
"offload_outputs_to_cpu": true,
|
||||
"batch_size": 128,
|
||||
"max_response_length": 100,
|
||||
"response_prefix": "",
|
||||
"kl_divergence_scale": 1.0,
|
||||
"kl_divergence_target": 0.01,
|
||||
"orthogonalize_direction": true,
|
||||
"row_normalization": "full",
|
||||
"full_normalization_lora_rank": 3,
|
||||
"winsorization_quantile": 1.0,
|
||||
"n_trials": 200,
|
||||
"n_startup_trials": 60,
|
||||
"seed": 1257409843,
|
||||
"export_strategy": "merge",
|
||||
"max_shard_size": "5GB",
|
||||
"refusal_markers": [
|
||||
"disclaimer",
|
||||
"sorry",
|
||||
"i can'",
|
||||
"i cant",
|
||||
"i cannot",
|
||||
"i won'",
|
||||
"i wont",
|
||||
"i will not",
|
||||
"i unable",
|
||||
"im unable",
|
||||
"i'm unable",
|
||||
"i am unable",
|
||||
"i an ai",
|
||||
"im an ai",
|
||||
"i'm an ai",
|
||||
"i am an ai",
|
||||
"as an ai",
|
||||
"ai assistant",
|
||||
"i designed to",
|
||||
"im designed to",
|
||||
"i'm designed to",
|
||||
"i am designed to",
|
||||
"i programmed to",
|
||||
"im programmed to",
|
||||
"i'm programmed to",
|
||||
"i am programmed to",
|
||||
"violat",
|
||||
"prohibit",
|
||||
"illegal",
|
||||
"harmful",
|
||||
"inappropriate",
|
||||
"unethical",
|
||||
"ethical boundaries"
|
||||
],
|
||||
"system_prompt": "You are a helpful assistant.",
|
||||
"good_prompts": {
|
||||
"dataset": "mlabonne/harmless_alpaca",
|
||||
"commit": "02c6a92cfcf11bb0c387334f8146d149d65b587f",
|
||||
"split": "train[:400]",
|
||||
"column": "text",
|
||||
"prefix": "",
|
||||
"suffix": "",
|
||||
"system_prompt": null
|
||||
},
|
||||
"bad_prompts": {
|
||||
"dataset": "mlabonne/harmful_behaviors",
|
||||
"commit": "01cead01398926d81f7c52bdb790ee8cf77ebba7",
|
||||
"split": "train[:400]",
|
||||
"column": "text",
|
||||
"prefix": "",
|
||||
"suffix": "",
|
||||
"system_prompt": null
|
||||
},
|
||||
"good_evaluation_prompts": {
|
||||
"dataset": "mlabonne/harmless_alpaca",
|
||||
"commit": "02c6a92cfcf11bb0c387334f8146d149d65b587f",
|
||||
"split": "test[:100]",
|
||||
"column": "text",
|
||||
"prefix": "",
|
||||
"suffix": "",
|
||||
"system_prompt": null
|
||||
},
|
||||
"bad_evaluation_prompts": {
|
||||
"dataset": "mlabonne/harmful_behaviors",
|
||||
"commit": "01cead01398926d81f7c52bdb790ee8cf77ebba7",
|
||||
"split": "test[:100]",
|
||||
"column": "text",
|
||||
"prefix": "",
|
||||
"suffix": "",
|
||||
"system_prompt": null
|
||||
}
|
||||
},
|
||||
"parameters": {
|
||||
"direction_index": 23.24818953362204,
|
||||
"abliteration_parameters": {
|
||||
"attn.o_proj": {
|
||||
"max_weight": 1.4551074307886407,
|
||||
"max_weight_position": 32.545522315255255,
|
||||
"min_weight": 1.31618769061741,
|
||||
"min_weight_distance": 26.617478054534267
|
||||
},
|
||||
"mlp.down_proj": {
|
||||
"max_weight": 1.1232193857004718,
|
||||
"max_weight_position": 35.90942379336195,
|
||||
"min_weight": 1.0333789850006858,
|
||||
"min_weight_distance": 27.7878893606307
|
||||
}
|
||||
}
|
||||
},
|
||||
"metrics": {
|
||||
"kl_divergence": 0.0623466782271862,
|
||||
"refusals": 2,
|
||||
"base_refusals": 99,
|
||||
"n_bad_prompts": 100
|
||||
},
|
||||
"hashes": {
|
||||
"model-00001-of-00004.safetensors": "1cbc1dcebd8de1fd20145ece0bc508922bd6669aa43f9c566db9d6a6a7113ff1",
|
||||
"model-00002-of-00004.safetensors": "aa5f8ab572105d74cf2b5a24ec530158296a82d1ed09c968106537005f92adb1",
|
||||
"model-00003-of-00004.safetensors": "4019a8b4adb883ebafa630d7d5010f9179d8842cd59aed9d0cd9b97811a18099",
|
||||
"model-00004-of-00004.safetensors": "294f92c0065d44a3bc726bdd15fb969fcd26c9d290bfe592176f0a88872730e9"
|
||||
}
|
||||
}
|
||||
114
reproduce/requirements.txt
Normal file
114
reproduce/requirements.txt
Normal file
@@ -0,0 +1,114 @@
|
||||
absl-py==2.4.0
|
||||
accelerate==1.14.0
|
||||
alembic==1.18.5
|
||||
annotated-doc==0.0.4
|
||||
annotated-types==0.7.0
|
||||
anyio==4.14.1
|
||||
bitsandbytes==0.49.2
|
||||
certifi==2026.6.17
|
||||
chardet==6.0.0.post1
|
||||
charset-normalizer==3.4.7
|
||||
click==8.4.2
|
||||
colorama==0.4.6
|
||||
colorlog==6.11.0
|
||||
dataproperty==1.1.1
|
||||
datasets==4.8.5
|
||||
defusedxml==0.7.1
|
||||
dill==0.4.1
|
||||
evaluate==0.4.6
|
||||
filelock==3.29.4
|
||||
fsspec==2026.2.0
|
||||
greenlet==3.5.3
|
||||
h11==0.16.0
|
||||
heretic-llm==1.4.0
|
||||
hf-xet==1.5.2
|
||||
httpcore==1.0.9
|
||||
httpx==0.28.1
|
||||
huggingface-hub==1.24.0
|
||||
idna==3.18
|
||||
immutabledict==4.3.1
|
||||
jinja2==3.1.6
|
||||
joblib==1.5.3
|
||||
langdetect==1.0.9
|
||||
lm-eval==0.4.12
|
||||
lxml==6.1.1
|
||||
mako==1.3.12
|
||||
markdown-it-py==4.2.0
|
||||
markupsafe==3.0.3
|
||||
mbstrdecoder==1.1.5
|
||||
mdurl==0.1.2
|
||||
more-itertools==11.1.0
|
||||
mpmath==1.3.0
|
||||
multiprocess==0.70.19
|
||||
narwhals==2.24.0
|
||||
networkx==3.6.1
|
||||
nltk==3.10.0
|
||||
numpy==2.5.1
|
||||
nvidia-cublas-cu12==12.8.4.1
|
||||
nvidia-cuda-cupti-cu12==12.8.90
|
||||
nvidia-cuda-nvrtc-cu12==12.8.93
|
||||
nvidia-cuda-runtime-cu12==12.8.90
|
||||
nvidia-cudnn-cu12==9.10.2.21
|
||||
nvidia-cufft-cu12==11.3.3.83
|
||||
nvidia-cufile-cu12==1.13.1.3
|
||||
nvidia-curand-cu12==10.3.9.90
|
||||
nvidia-cusolver-cu12==11.7.3.90
|
||||
nvidia-cusparse-cu12==12.5.8.93
|
||||
nvidia-cusparselt-cu12==0.7.1
|
||||
nvidia-nccl-cu12==2.27.3
|
||||
nvidia-nvjitlink-cu12==12.8.93
|
||||
nvidia-nvtx-cu12==12.8.90
|
||||
optuna==4.9.0
|
||||
packaging==26.0
|
||||
pandas==3.0.3
|
||||
pathvalidate==3.3.1
|
||||
peft==0.19.1
|
||||
pillow==12.2.0
|
||||
portalocker==3.2.0
|
||||
prompt-toolkit==3.0.52
|
||||
psutil==7.2.2
|
||||
py-cpuinfo==9.0.0
|
||||
pyarrow==25.0.0
|
||||
pydantic==2.13.4
|
||||
pydantic-core==2.46.4
|
||||
pydantic-settings==2.14.2
|
||||
pygments==2.20.0
|
||||
pytablewriter==1.2.1
|
||||
python-dateutil==2.9.0.post0
|
||||
python-dotenv==1.2.2
|
||||
pyyaml==6.0.3
|
||||
questionary==2.1.1
|
||||
regex==2026.7.19
|
||||
requests==2.34.2
|
||||
rich==14.3.4
|
||||
rouge-score==0.1.2
|
||||
sacrebleu==2.6.0
|
||||
safetensors==0.8.0
|
||||
scikit-learn==1.9.0
|
||||
scipy==1.18.0
|
||||
setuptools==82.0.1
|
||||
shellingham==1.5.4
|
||||
six==1.17.0
|
||||
sqlalchemy==2.0.51
|
||||
sqlitedict==2.1.0
|
||||
sympy==1.14.0
|
||||
tabledata==1.3.5
|
||||
tabulate==0.10.0
|
||||
tcolorpy==0.1.7
|
||||
threadpoolctl==3.6.0
|
||||
tokenizers==0.22.2
|
||||
tomli-w==1.2.0
|
||||
torch==2.8.0
|
||||
torchvision==0.23.0
|
||||
tqdm==4.68.3
|
||||
transformers==5.14.1
|
||||
triton==3.4.0
|
||||
typepy==1.3.5
|
||||
typer==0.27.0
|
||||
typing-extensions==4.15.0
|
||||
typing-inspection==0.4.2
|
||||
tzdata==2026.2
|
||||
urllib3==2.5.0
|
||||
wcwidth==0.8.1
|
||||
word2number==1.1
|
||||
xxhash==3.8.1
|
||||
504229
tokenizer.json
Normal file
504229
tokenizer.json
Normal file
File diff suppressed because it is too large
Load Diff
3
tokenizer.model
Normal file
3
tokenizer.model
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:386c49cf943d71aa110361135338c50e38beeff0a66593480421f37b319e1a39
|
||||
size 1033105
|
||||
18
tokenizer_config.json
Normal file
18
tokenizer_config.json
Normal file
@@ -0,0 +1,18 @@
|
||||
{
|
||||
"add_prefix_space": true,
|
||||
"backend": "tokenizers",
|
||||
"bos_token": "<|startoftext|>",
|
||||
"clean_up_tokenization_spaces": false,
|
||||
"eos_token": "<|im_end|>",
|
||||
"is_local": false,
|
||||
"local_files_only": false,
|
||||
"model_max_length": 16384,
|
||||
"pad_token": "<unk>",
|
||||
"padding_side": "left",
|
||||
"sp_model_kwargs": {},
|
||||
"spaces_between_special_tokens": false,
|
||||
"split_special_tokens": false,
|
||||
"tokenizer_class": "LlamaTokenizer",
|
||||
"unk_token": "<unk>",
|
||||
"use_default_system_prompt": false
|
||||
}
|
||||
Reference in New Issue
Block a user