Update README.md (#3)
Browse files- Update README.md (08f3af2ab73ca8b5702ae0547afc7b7adb9c3b97)
Co-authored-by: Aju Vattakunnel <[email protected]>
README.md
CHANGED
|
@@ -170,6 +170,7 @@ pip3 install git+https://github.com/huggingface/transformers.git
|
|
| 170 |
|
| 171 |
```python
|
| 172 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
|
|
|
| 173 |
|
| 174 |
model_name_or_path = "TheBloke/Asclepius-13B-GPTQ"
|
| 175 |
# To use a different branch, change revision
|
|
|
|
| 170 |
|
| 171 |
```python
|
| 172 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
| 173 |
+
import torch
|
| 174 |
|
| 175 |
model_name_or_path = "TheBloke/Asclepius-13B-GPTQ"
|
| 176 |
# To use a different branch, change revision
|