Views
No views yet
1models:
2 - model: automerger/YamShadow-7B
3 parameters:
4 density: 0.5
5 weight: 0.5
6 - model: yam-peleg/Experiment28-7B
7 parameters:
8 density: 0.5
9 weight: 0.5
10merge_method: dare_linear
11base_model: automerger/YamShadow-7B
12dtype: float16
13random_seed: 0
14 ```
15## 💻 Usage
16```python
17!pip install -qU transformers accelerate
18from transformers import AutoTokenizer
19import transformers
20import torch
21model = "aaron-di/YamshadowExperiment28-7B-DareLinear"
22messages = [{"role": "user", "content": "What is a large language model?"}]
23tokenizer = AutoTokenizer.from_pretrained(model)
24prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
25pipeline = transformers.pipeline(
26 "text-generation",
27 model=model,
28 torch_dtype=torch.float16,
29 device_map="auto",
30)
31outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
32print(outputs[0]["generated_text"])