Update README.md
Browse files
README.md
CHANGED
@@ -30,4 +30,35 @@ parameters:
|
|
30 |
value: [1, 0.5, 0.7, 0.3, 0]
|
31 |
- value: 0.5
|
32 |
dtype: bfloat16
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
```
|
|
|
30 |
value: [1, 0.5, 0.7, 0.3, 0]
|
31 |
- value: 0.5
|
32 |
dtype: bfloat16
|
33 |
+
```
|
34 |
+
|
35 |
+
## 💻 Usage
|
36 |
+
|
37 |
+
```python
|
38 |
+
!pip install -qU transformers accelerate
|
39 |
+
|
40 |
+
from transformers import AutoTokenizer
|
41 |
+
import transformers
|
42 |
+
import torch
|
43 |
+
|
44 |
+
model = "mlabonne/NeuralPipe-7B-slerp"
|
45 |
+
messages = [{"role": "user", "content": "What is a large language model?"}]
|
46 |
+
|
47 |
+
tokenizer = AutoTokenizer.from_pretrained(model)
|
48 |
+
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
49 |
+
pipeline = transformers.pipeline(
|
50 |
+
"text-generation",
|
51 |
+
model=model,
|
52 |
+
torch_dtype=torch.float16,
|
53 |
+
device_map="auto",
|
54 |
+
)
|
55 |
+
|
56 |
+
outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
|
57 |
+
print(outputs[0]["generated_text"])
|
58 |
+
```
|
59 |
+
|
60 |
+
Output:
|
61 |
+
|
62 |
+
```
|
63 |
+
A large language model is an AI system that uses deep learning techniques to process and understand vast amounts of natural language data. It is designed to generate human-like text, perform complex language tasks, and understand the context, nuance, and meaning of textual data. These models are trained on large datasets, often including billions of words, to learn the patterns and relationships in language. As a result, they can generate coherent and contextually relevant text, answer questions, and perform a variety of other language-related tasks. Some well-known large language models include OpenAI's GPT-3, Google's BERT, and Facebook's RoBERTa.
|
64 |
```
|