Commit
•
90c88db
1
Parent(s):
0218dec
Update README.md (#1)
Browse files- Update README.md (d9d4a224b1d6ebe75555e4e7c2bf6311d1f730bc)
Co-authored-by: Ravi Theja <[email protected]>
README.md
CHANGED
|
@@ -128,8 +128,7 @@ response = tokenizer.batch_decode(outputs)
|
|
| 128 |
# Inference with HuggingFace
|
| 129 |
|
| 130 |
```python3
|
| 131 |
-
from
|
| 132 |
-
from transformers import AutoTokenizer
|
| 133 |
import torch
|
| 134 |
|
| 135 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 128 |
# Inference with HuggingFace
|
| 129 |
|
| 130 |
```python3
|
| 131 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
|
| 132 |
import torch
|
| 133 |
|
| 134 |
model = AutoModelForCausalLM.from_pretrained(
|