Update README.md (#3)
Browse files- Update README.md (08f3af2ab73ca8b5702ae0547afc7b7adb9c3b97)
Co-authored-by: Aju Vattakunnel <avr266@users.noreply.huggingface.co>
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
|