Update README.md
Browse files
README.md
CHANGED
|
@@ -17,6 +17,11 @@ base_model: Viet-Mistral/Vistral-7B-Chat
|
|
| 17 |
This model was fine-tuned on Vistral-7B-chat for function calling.
|
| 18 |
|
| 19 |
## Usage
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
```python
|
| 21 |
import torch
|
| 22 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
| 17 |
This model was fine-tuned on Vistral-7B-chat for function calling.
|
| 18 |
|
| 19 |
## Usage
|
| 20 |
+
|
| 21 |
+
```sh
|
| 22 |
+
pip install accelerate
|
| 23 |
+
```
|
| 24 |
+
|
| 25 |
```python
|
| 26 |
import torch
|
| 27 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|