Fix import in README.md
#4
by
rusch
- opened
README.md
CHANGED
|
@@ -24,6 +24,7 @@ datasets:
|
|
| 24 |
Get started chatting with `StableLM-Tuned-Alpha` by using the following code snippet:
|
| 25 |
|
| 26 |
```python
|
|
|
|
| 27 |
from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList
|
| 28 |
|
| 29 |
tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
|
|
|
|
| 24 |
Get started chatting with `StableLM-Tuned-Alpha` by using the following code snippet:
|
| 25 |
|
| 26 |
```python
|
| 27 |
+
import torch
|
| 28 |
from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList
|
| 29 |
|
| 30 |
tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
|