CultriX commited on
Commit
59ae468
1 Parent(s): 6d10d02

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +72 -0
README.md ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - merge
4
+ - mergekit
5
+ - lazymergekit
6
+ - FelixChao/WestSeverus-7B-DPO-v2
7
+ - CultriX/Wernicke-7B-v8
8
+ - vanillaOVO/supermario_v2
9
+ base_model:
10
+ - FelixChao/WestSeverus-7B-DPO-v2
11
+ - CultriX/Wernicke-7B-v8
12
+ - vanillaOVO/supermario_v2
13
+ license: apache-2.0
14
+ ---
15
+
16
+ # Edit:
17
+ * DPO-finetune using truthy-dpo dataset and CultriX/Wernicke-7B-v9.
18
+
19
+ # Wernicke-7B-v9
20
+
21
+ Wernicke-7B-v9 is a merge of the following models using [LazyMergekit](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb?usp=sharing):
22
+ * [FelixChao/WestSeverus-7B-DPO-v2](https://huggingface.co/FelixChao/WestSeverus-7B-DPO-v2)
23
+ * [CultriX/Wernicke-7B-v8](https://huggingface.co/CultriX/Wernicke-7B-v8)
24
+ * [vanillaOVO/supermario_v2](https://huggingface.co/vanillaOVO/supermario_v2)
25
+
26
+ ## 🧩 Configuration
27
+
28
+ ```yaml
29
+ models:
30
+ - model: FelixChao/WestSeverus-7B-DPO-v2
31
+ parameters:
32
+ density: 0.50
33
+ weight: 0.35
34
+ - model: CultriX/Wernicke-7B-v8
35
+ parameters:
36
+ density: 0.50
37
+ weight: 0.35
38
+ - model: vanillaOVO/supermario_v2
39
+ parameters:
40
+ density: 0.50
41
+ weight: 0.30
42
+ merge_method: dare_ties
43
+ base_model: FelixChao/WestSeverus-7B-DPO-v2
44
+ parameters:
45
+ int8_mask: true
46
+ dtype: float16
47
+ ```
48
+
49
+ ## 💻 Usage
50
+
51
+ ```python
52
+ !pip install -qU transformers accelerate
53
+
54
+ from transformers import AutoTokenizer
55
+ import transformers
56
+ import torch
57
+
58
+ model = "CultriX/Wernicke-7B-v9"
59
+ messages = [{"role": "user", "content": "What is a large language model?"}]
60
+
61
+ tokenizer = AutoTokenizer.from_pretrained(model)
62
+ prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
63
+ pipeline = transformers.pipeline(
64
+ "text-generation",
65
+ model=model,
66
+ torch_dtype=torch.float16,
67
+ device_map="auto",
68
+ )
69
+
70
+ outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
71
+ print(outputs[0]["generated_text"])
72
+ ```