Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,6 @@ The following hyperparameters were used during training:
|
|
39 |
|
40 |
### Training results
|
41 |
|
42 |
-
|
43 |
### Framework versions
|
44 |
|
45 |
- AutoAWQ 0.1.8
|
@@ -47,4 +46,42 @@ The following hyperparameters were used during training:
|
|
47 |
- Transformers 4.37.0.dev0
|
48 |
- Pytorch 2.1.2+cu121
|
49 |
- Datasets 2.16.2.dev0
|
50 |
-
- Tokenizers 0.15.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
|
40 |
### Training results
|
41 |
|
|
|
42 |
### Framework versions
|
43 |
|
44 |
- AutoAWQ 0.1.8
|
|
|
46 |
- Transformers 4.37.0.dev0
|
47 |
- Pytorch 2.1.2+cu121
|
48 |
- Datasets 2.16.2.dev0
|
49 |
+
- Tokenizers 0.15.0
|
50 |
+
|
51 |
+
|
52 |
+
## Example usage
|
53 |
+
|
54 |
+
Using `transformers` and `AutoAWQ`:
|
55 |
+
|
56 |
+
```shell
|
57 |
+
pip install -U transformers autoawq
|
58 |
+
```
|
59 |
+
|
60 |
+
```
|
61 |
+
# Use a pipeline as a high-level helper
|
62 |
+
from transformers import pipeline
|
63 |
+
|
64 |
+
messages = [
|
65 |
+
{"role": "system", "content": "You are an helpful Assistant."},
|
66 |
+
{"role": "user", "content": "Who are you?"},
|
67 |
+
]
|
68 |
+
pipe = pipeline("text-generation", model="wasertech/assistant-dolphin-2.2.1-mistral-7b-e1-awq", max_length=8096)
|
69 |
+
pipe(messages)
|
70 |
+
```
|
71 |
+
|
72 |
+
Outputs
|
73 |
+
|
74 |
+
```
|
75 |
+
[
|
76 |
+
{
|
77 |
+
'generated_text': [
|
78 |
+
{'role': 'system', 'content': 'You are an helpful Assistant.'},
|
79 |
+
{'role': 'user', 'content': 'Who are you?'},
|
80 |
+
{'role': 'assistant', 'content': '<|im_start|> Assistant\nI am an artificial intelligence language model. My purpose is to provide information, advice, and assistance to users. I can perform many tasks, such as answering questions, explaining concepts, generating reports, or summarizing data. I am a tool that can be used to help you learn new things, make decisions, and achieve your goals. I do not have feelings, opinions, or personal experiences; I am simply a helpful resource to share knowledge and support users. What can I do for you?'}
|
81 |
+
]
|
82 |
+
}
|
83 |
+
]
|
84 |
+
```
|
85 |
+
|
86 |
+
Parsed Assistant answer:
|
87 |
+
> I am an artificial intelligence language model. My purpose is to provide information, advice, and assistance to users. I can perform many tasks, such as answering questions, explaining concepts, generating reports, or summarizing data. I am a tool that can be used to help you learn new things, make decisions, and achieve your goals. I do not have feelings, opinions, or personal experiences; I am simply a helpful resource to share knowledge and support users. What can I do for you?
|